Update app.py

app.py

@@ -32,7 +32,7 @@ def predict(image):
         
         engResult = clientEngText.predict(
 		text=ocr_text,
-		api_name="/classify_text"
+		api_name="/predict"
         )
         
         hingResult = clientHingText.predict(
@@ -44,9 +44,9 @@ def predict(image):
 
         if profanityFound:
             return ["hate", "Profanity Found"]
-        elif engResult[0] != "NEITHER":
+        elif engResult[0] != "NEITHER" or engResult[1] < 0.9:
             return ["hate", f"Result: {engResult}, Text: {ocr_text}"]
-        elif hingResult[0] != "NAG":
+        elif hingResult[0] != "NAG" or hingResult:
             return ["hate", f"Result: {hingResult}, Text: {ocr_text}"]
         else:
             return ["not_hate", "No hate found, yay!"]
@@ -58,7 +58,7 @@ def predict(image):
         
 iface = gr.Interface(fn=predict,
                      inputs = gr.Image(type='filepath'),
-                     outputs=[gr.Label(label = "Class (Hate or Not Hate)") , gr.Label(label = "Hate Symbol (if any)")],
+                     outputs=[gr.Label(label = "Class (hate or not_hate)") , gr.Label(label = "Explanation")],
                      title = "Hate Speech Detection in Image",
                      description = "Detect hateful symbols or text in Image"
         )