Update api.py

api.py

@@ -5,7 +5,25 @@ import io
 import tensorflow as tf
 
 
-model = tf.keras.models.load_model('_9217')
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+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)
+  outputs = model(**inputs)
+
+  # Assuming the model outputs logits
+  predicted_class = outputs.logits.argmax(dim=-1).item()
+
+  class_labels = [
+      'Improvement Suggestions', 'Questions', 'Confusion', 'Support Request',
+      'Discussion', 'Course Comparison', 'Related Course Suggestions',
+      'Negative', 'Positive'
+  ]
+  return class_labels[predicted_class]
+
 
 app = FastAPI()
 @app.post("/classify")