Modified app.py

app.py

@@ -128,7 +128,7 @@ def predict(tweet):
         if type(prediction) == dict: 
             print(prediction)
             error_message = prediction['error']
-            return error_message, [[]]
+            return error_message, []
         
         prediction = prediction[0][0]["label"]
         
@@ -136,9 +136,9 @@ def predict(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 +146,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,5 +159,4 @@ demo = gr.Interface(
     title="Tagalog Profanity Classifier"
 )
 
-demo.launch(debug=True)
-predict("Tangina mo naman gag0 ka ba")
+demo.launch(debug=True)