Update app.py

app.py

@@ -15,7 +15,7 @@ app = FastAPI()
 pipe = pipeline(
     "image-to-text",
     model="HuggingFaceTB/SmolVLM-256M-Instruct",
-    device=-1  # CPU (works on free tier)
+    device=-1  # CPU for free tier
 )
 
 @app.get("/")
@@ -41,20 +41,21 @@ def upload_form():
 @app.post("/predict")
 async def predict_gender(file: UploadFile = File(...)):
     try:
-        # Read image
+        # Read uploaded image
         image_bytes = await file.read()
         image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
 
-        # Ask model to classify gender
+        # Instruction for the model
         prompt = "Is the person on this ID male or female?"
-        result = pipe(image, prompt, max_new_tokens=32)
 
-        # Extract answer
+        # Run model (pipeline handles image + prompt via generate_kwargs)
+        result = pipe(image, generate_kwargs={"max_new_tokens": 32, "prompt": prompt})
+
+        # Extract model output
         answer = result[0]["generated_text"].strip()
 
         return JSONResponse({"gender": answer})
 
     except Exception as e:
-        # Print error to logs
         traceback.print_exc()
         return JSONResponse({"error": str(e)}, status_code=500)