Update app.py

app.py

@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 
-def predict(username, pwd, payload_text = None, label):
+def predict(username, pwd, label, payload_text = None):
     if(payload_text is None):
         payload["username"] = username
         payload["password"] = pwd
@@ -27,10 +27,10 @@ def predict(username, pwd, payload_text = None, label):
 demo = gr.Interface(
     fn=predict,
     inputs=[gr.Textbox(label="Enter Username"),gr.Textbox(label="Enter Password"),
-            gr.Textbox(label="Enter Payload[optional]"),
            gr.Dropdown(
             ["Malitious", "Benign"], label="Expected", info="Enter expected value"
-        )],
+        ),
+            gr.Textbox(label="Enter Payload[optional]")]
     outputs=[gr.Textbox(label="Model Prediction")]
 )
 demo.launch(debug=True)