Update api.py

api.py

@@ -5,10 +5,21 @@ import io
 import tensorflow as tf
 
 
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+# from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+# tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
+# model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
+
+
+from transformers import DistilBertTokenizerFast
+
+tokenizer = DistilBertTokenizerFast.from_pretrained("distilbert-base-uncased")
+
+from transformers import pipeline
+
+model = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
+
 
-tokenizer = AutoTokenizer.from_pretrained("chillies/distilbert-course-review-classification")
-model = AutoModelForSequenceClassification.from_pretrained("chillies/distilbert-course-review-classification")
 
 def inference(review):
   inputs = tokenizer(review, return_tensors="pt", padding=True, truncation=True)