fixing endpoint parameters

main.py

@@ -8,8 +8,16 @@ from fastapi import FastAPI, File, UploadFile, HTTPException
 from qwen_vl_utils import process_vision_info
 from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, Qwen2VLForConditionalGeneration
 
+from fastapi import FastAPI, Body
+from pydantic import BaseModel
+
 app = FastAPI()
 
+# Define request model
+class PredictRequest(BaseModel):
+    image_base64: str
+    prompt: str
+
 checkpoint = "Qwen/Qwen2-VL-2B-Instruct"
 min_pixels = 256 * 28 * 28
 max_pixels = 1280 * 28 * 28
@@ -86,7 +94,7 @@ async def upload_and_encode_image(file: UploadFile = File(...)):
         raise HTTPException(status_code=400, detail=f"Invalid file: {e}")
 
 @app.post("/predict")
-def predict(data: any = Query(...)):
+def predict(data: PredictRequest):
     """
     Generates a description for an image using the Qwen-2-VL model.
 

model.py

@@ -28,7 +28,8 @@ predict_payload = {
     "prompt": "describe the image",
 }
 
-predict_response = requests.post(f"{BASE_URL}/predict", params=predict_payload)
+predict_response = requests.post(f"{BASE_URL}/predict", json=predict_payload)
+
 
 # Step 4: Print the response
 if predict_response.status_code == 200: