Update app.py

app.py

@@ -27,12 +27,9 @@ def predict(username, pwd, label, payload_text = None):
     # logits = model(**inputs).logits
     # predicted_class_id = int(tf.math.argmax(logits, axis=-1)[0])
     # print(model.config.id2label[predicted_class_id])
-    prediction = classifier(payload_text)    
-    scores = {
-        prediction[0]['label']: prediction[0]['score'] * 100
-
-    }
-    return payload_text, scores
+    prediction = classifier(payload_text)[0]    
+    
+    return payload_text, prediction
 
 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")]