Update app.py

app.py

@@ -24,7 +24,9 @@ def predict(image):
     
     label , confidence = imgResult[0]['label'] , float(imgResult[1]['label'])
 
-    if confidence > 0.95:
+    if label == 'finger_gun_to_the_head' and confidence > 0.98:
+        return ["hate" , f"label: {label}, confidence: {confidence}"]
+    elif label != 'finger_gun_to_the_head' and confidence > 0.95:
         return ["hate" , f"label: {label}, confidence: {confidence}"]
     else: