Update app.py

app.py

@@ -1,3 +1,17 @@
 import gradio as gr
+def predict(payload, payload_label):
+    inputs = tokenizer(test_payload, return_tensors="tf")
+    model = TFAutoModelForSequenceClassification.from_pretrained("models/ankush-003/nosqli_identifier")
+    logits = model(**inputs).logits
+    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 "Fuck You"
 
-gr.Interface.load("models/ankush-003/nosqli_identifier").launch()
+demo = gr.Interface(
+    fn=predict,
+    inputs=["text","checkbox"],
+    outputs="text",
+)
+demo.launch(share=True, debug=True)
+# gr.Interface.load("models/ankush-003/nosqli_identifier").launch()