Fix error handling issues

app.py

@@ -117,29 +117,30 @@ def query(payload):
 
 
 def predict(text):
+    output = query(preprocess(text))
     print(preprocess(text))
     
-    try:
-        output = query(preprocess(text))[0]
-    except Exception:
-        return "Loading", "The model is loading please try again later."
-
-    output = [tuple(i.values()) for i in output]
-    output = dict((x, y) for x, y in output)
-
-    predicted_label = list(output.keys())[0]
-
-    if predicted_label == 'Abusive':
-        output_text = text
-        for profanity in profanities:
-            compiled = re.compile(re.escape(profanity), re.IGNORECASE)
-            mask = ""
-            for i in profanity:
-                mask += "*" if i != " " else " "
-            output_text = compiled.sub(mask, output_text)
-        return output, output_text
+    if 'error' in output:
+        return output['error'], 'Error occured. Try again later.'
     else:
-        return output, text
+        output = [tuple(i.values()) for i in output[0]]
+        output = dict((x, y) for x, y in output)
+
+        predicted_label = list(output.keys())[0]
+
+        if predicted_label == 'Abusive':
+            output_text = text
+            for profanity in profanities:
+                compiled = re.compile(re.escape(profanity), re.IGNORECASE)
+                mask = ""
+                for i in profanity:
+                    mask += "*" if i != " " else " "
+                output_text = compiled.sub(mask, output_text)
+            return output, output_text
+        else:
+            return output, text
+    
+    
 
 
 hf_writer = gr.HuggingFaceDatasetSaver('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', 'tagalog-profanity-feedbacks')