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)    
     
-    return payload_text, prediction['label'], prediction['score'] * 100
+    return payload_text, prediction[0]['label'], prediction[0]['score'] * 100
 
 input_elements = [gr.Textbox(label="Enter Username"), gr.Textbox(label="Enter Password"), gr.Dropdown(["Malitious", "Benign"], label="Expected", info="Enter expected value"),
             gr.Textbox(label="Enter Payload", info="Optional if username and password entered already")]