Update app.py

app.py

@@ -12,7 +12,7 @@ tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 
 def predict(username, pwd, label, payload_text = None):
-    if(payload_text is None):
+    if(payload_text is None or payload_text = ""):
         payload["username"] = username
         payload["password"] = pwd
         payload_text = json.dumps(payload)
@@ -22,7 +22,7 @@ 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], payload_text
+    return payload_text, 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", info="Optional if username and password entered already")]