Update main.py

main.py

@@ -173,7 +173,7 @@ class Guard :
             print('-'*100)
             print(f'THE SCORE ATTACK WITH BAYES IS {score} and the SCORES WITH LOGIT ATT:{score2} DEF:{score3} ')
             print('-'*100)
-
+            
             f= 0.5*score + 0.5*score2 
             if score3>f and score3>0.9: 
                 prompt_info['is_safe'] = True
@@ -181,7 +181,7 @@ class Guard :
                 prompt_info['confidence_score'] = 100*round(score3, 4)
                 prompt_info['latency'] = round(time.time() - start_time , 4)
                 return prompt_info
-            elif score3 <0.5: 
+            elif score3 <0.5 and score2 >0.4: 
                 prompt_info['is_safe'] = False 
                 prompt_info['decision'] = 'BLOCKED'
                 prompt_info['confidence_score'] = 100