Update api.py

api.py

@@ -69,17 +69,21 @@ def inference(review):
   ]
   return class_labels[predicted_class]
 
+from pydantic import BaseModel
+from typing import List
+
+class ReviewRequest(BaseModel):
+    reviews: List[str]
 
 app = FastAPI()
 @app.post("/classify")
-async def classify(request):
-    print("HERE",request)
+async def classify(request: ReviewRequest):
+    print("HERE", request)
     reviews = request.reviews
     predictions = []
 
-    # Process each review and get the predictions
     for review in reviews:
         predicted_class = inference(review)
-        predictions.append({predicted_class})
-        
+        predictions.append(predicted_class)
+
     return {"predictions": predictions}