Update app.py

app.py

@@ -24,13 +24,12 @@ def predict(username, pwd, label, payload_text = None):
     
     return model.config.id2label[predicted_class_id]
 
+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]")]
+
 demo = gr.Interface(
     fn=predict,
-    inputs=[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]")]
+    inputs=input_elements,
     outputs=[gr.Textbox(label="Model Prediction")]
 )
 demo.launch(debug=True)