Update app.py

app.py

@@ -22,15 +22,15 @@ def predict(username, pwd, label, payload_text = None):
     predicted_class_id = int(tf.math.argmax(logits, axis=-1)[0])
     # print(model.config.id2label[predicted_class_id])
     
-    return model.config.id2label[predicted_class_id]
+    return model.config.id2label[predicted_class_id], payload_text
 
 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[optional]")]
+            gr.Textbox(label="Enter Payload", info="Optional if username and password entered already")]
 
 demo = gr.Interface(
     fn=predict,
     inputs=input_elements,
-    outputs=[gr.Textbox(label="Model Prediction")]
+    outputs=[gr.Textbox(label="Generated Payload"), gr.Textbox(label="Model Prediction")]
 )
 demo.launch(debug=True)
 # gr.Interface.load("models/ankush-003/nosqli_identifier").launch()