Upload app.py

app.py

@@ -91,14 +91,9 @@ async def predict_age_gender(file: UploadFile = File(...)):
         if img is None:
             raise HTTPException(status_code=400, detail="Invalid or unreadable image file.")
 
-        # Write to temp file so MiVOLO can read it from disk
-        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
-            tmp_path = tmp.name
-            cv2.imwrite(tmp_path, img)
-
-        # Run MiVOLO prediction
+        # Run MiVOLO prediction directly on the numpy image array
         pred = get_predictor()
-        detected_objects, _ = pred.recognize(tmp_path)
+        detected_objects, _ = pred.recognize(img)
 
         if detected_objects is None or not detected_objects.ages:
             raise HTTPException(