Update app.py

app.py

@@ -8,7 +8,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 # tokenizer = AutoTokenizer.from_pretrained("ankush-003/nosqli_identifier")
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
-model = AutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
+model = TFAutoModelForSequenceClassification.from_pretrained("ankush-003/nosqli_identifier")
 
 def predict(payload, malitious):
     inputs = tokenizer(payload, return_tensors="tf")