Update app.py

app.py

@@ -29,7 +29,7 @@ def predict(username, pwd, label, payload_text = None):
     # print(model.config.id2label[predicted_class_id])
     prediction = classifier(payload_text)[0]    
     
-    return payload_text, prediction
+    return payload_text, {prediction["label"]: prediction["score"]}
 
 input_elements = [gr.Textbox(label="Enter Username"), gr.Textbox(label="Enter Password"), gr.Dropdown(["Malicious", "Benign"], label="Expected", info="Enter expected value"),
             gr.Textbox(label="Enter Payload", info="Optional if username and password entered already")]