Update main.py

main.py

@@ -4,7 +4,7 @@ from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 from ml_service import MlProcessing
 from ml_service import process_single_comment
-from typing import List, Dict
+from typing import List, Dict, Union
 import time
 
 app = FastAPI()
@@ -14,7 +14,7 @@ class TextRatingRequest(BaseModel):
     rating: int
 
 class AllTextRatingRequest(BaseModel):
-    reviews: List[Dict[str, float]]
+    reviews: List[Dict[str, Union[str, float, int]]
 
 @app.post("/predict")
 def predict_single_review(text_rating: TextRatingRequest):