Update app.py

app.py

@@ -18,12 +18,14 @@ def extract_text(image):
 def predict(image):
     
     imgResult = clientImg.predict(
-        image,
+        image=handle_file(image),
         api_name="/predict"
     )
     
-    if float(imgResult[1]['label']) > 0.95:
-        return ["hate" , imgResult[0]]
+    label , confidence = imgResult[0]['label'] , float(imgResult[1]['label'])
+
+    if confidence > 0.95:
+        return ["hate" , label]
     else:
  
         ocr_text = extract_text(image)
@@ -39,9 +41,9 @@ def predict(image):
         )
         
         if engResult[0] == "NEITHER" or hingResult[0] == "NAG":
-            return ["not_hate" , None]
+            return ["not_hate" , "No Hate Symbols Detected"]
         else:
-            return ["hate" , None]
+            return ["hate" , "No Hate Symbols Detected"]
         
 iface = gr.Interface(fn=predict,
                      inputs = gr.Image(type='filepath'),
@@ -52,4 +54,3 @@ iface = gr.Interface(fn=predict,
                      
 if __name__ == "__main__":
     iface.launch()
-