modified app.py

app.py

@@ -128,17 +128,16 @@ def predict(tweet):
         if type(prediction) == dict: 
             print(prediction)
             error_message = prediction['error']
-            return error_message, [[]]
-        
+            return error_message, {}
         prediction = prediction[0][0]["label"]
         
         print("\nTWEET:", tweet)
         print("DETECTED PROFANITY:", matched_profanity)
         print("LABEL:", prediction, "\n")
 
-        return prediction, [matched_profanity]
+        return prediction, matched_profanity
     
-    return "No Profanity", [[]]
+    return "No Profanity", {}
 
 
 demo = gr.Interface(
@@ -146,7 +145,7 @@ demo = gr.Interface(
 
     inputs=[gr.components.Textbox(lines=5, placeholder='Enter your input here', label='INPUT')],
 
-    outputs=[gr.components.Text(label="PREDICTION"), gr.List(label="PROFANITIES")],
+    outputs=[gr.components.Text(label="PREDICTION"), gr.JSON(label="PROFANITIES")],
 
     examples=['Tangina mo naman sobrang yabang mo gago!!😠😤 @davidrafael',
               'Napakainit ngayong araw pakshet namaaan!!',
@@ -159,4 +158,4 @@ demo = gr.Interface(
     title="Tagalog Profanity Classifier"
 )
 
-demo.launch(debug=True)
+demo.launch(debug=True)