Update api.py

api.py

@@ -27,15 +27,13 @@ def inference(review):
 
 app = FastAPI()
 @app.post("/classify")
-async def classify(image: UploadFile = File(...)):
-    if image is not None:
-        img = Image.open(io.BytesIO(await image.read()))
-        img = img.resize((128, 128))
-        img_array = np.array(img) / 255.0
-        img_array = np.expand_dims(img_array, axis=0)
-        predictions = model.predict(img_array)
-        predicted_class_idx = np.argmax(predictions)
-        predicted_class_idx = int(predicted_class_idx)
-        return {"prediction": predicted_class_idx}
-    else:
-        return {"error": "No image provided"}
+async def classify(request: ReviewRequest):
+    reviews = request.reviews
+    predictions = []
+
+    # Process each review and get the predictions
+    for review in reviews:
+        predicted_class = inference(review)
+        predictions.append({predicted_class})
+        
+    return {"predictions": predictions}