fixing model

main.py

@@ -10,7 +10,7 @@ from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, Qwen
 
 app = FastAPI()
 
-checkpoint = "Qwen/Qwen2-VL-7B-Instruct"
+checkpoint = "Qwen/Qwen2-VL-2B-Instruct"
 min_pixels = 256 * 28 * 28
 max_pixels = 1280 * 28 * 28
 processor = AutoProcessor.from_pretrained(
@@ -86,7 +86,7 @@ async def upload_and_encode_image(file: UploadFile = File(...)):
         raise HTTPException(status_code=400, detail=f"Invalid file: {e}")
 
 @app.get("/predict")
-def describe_image_with_qwen2_vl(image_url: str = Query(...), prompt: str = Query(...)):
+def predict(image_url: str = Query(...), prompt: str = Query(...)):
     """
     Generates a description for an image using the Qwen-2-VL model.