Update app.py

app.py

@@ -4,7 +4,7 @@ from moderation import *
 
 
 moderation = ModerationModel()
-moderation.load_state_dict(torch.load('moderation_model.pth', map_location=torch.device('cpu'))) #Remove map_location if run on gpu
+moderation.load_state_dict(torch.load('moderation_model.pth', map_location=torch.device('cpu'))) # Remove map_location if run on gpu
 moderation.eval()
 
 def predict_moderation(text):
@@ -17,10 +17,9 @@ def predict_moderation(text):
 
 iface = gr.Interface(fn=predict_moderation,
                      inputs="text",
-                     outputs=[gr.outputs.Label(label="Category Scores", type="confidences"),
-                              gr.outputs.Label(label="Detected")],
+                     outputs=[gr.Label(label="Category Scores"), gr.Label(label="Detected")],
                      title="Moderation Model",
                      description="Enter text to check for moderation flags.")
 
 
-iface.launch()
+iface.launch()