Update app.py

app.py

@@ -17,12 +17,13 @@ def predict(payload, malitious):
     predicted_class_id = int(tf.math.argmax(logits, axis=-1)[0])
     # print(model.config.id2label[predicted_class_id])
     expected = "Malitious" if malitious else "Benign"
+    
     return model.config.id2label[predicted_class_id], expected 
 
 demo = gr.Interface(
     fn=predict,
     inputs=["text","checkbox"],
-    outputs=["text","text"]
+    outputs=[gr.Textbox(label="Model Prediction"),gr.Textbox(label="Expected")]
 )
 demo.launch(debug=True)
 # gr.Interface.load("models/ankush-003/nosqli_identifier").launch()