Install error fix attemp 11

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, Form, HTTPException
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse
 from pydantic import BaseModel
 from PIL import Image
@@ -28,12 +28,12 @@ async def load_model():
     try:
         logger.info("Starting model loading...")
         
-        # Import required modules
-        from transformers import AutoProcessor, AutoModelForCausalLM
+        # Import required modules - use specific Qwen2VL classes
+        from transformers import Qwen2VLProcessor, Qwen2VLForConditionalGeneration
         
         logger.info("Loading processor...")
-        # Use AutoProcessor for better compatibility
-        processor = AutoProcessor.from_pretrained(
+        # Use specific Qwen2VL processor
+        processor = Qwen2VLProcessor.from_pretrained(
             model_name, 
             trust_remote_code=True
         )
@@ -42,8 +42,8 @@ async def load_model():
         tokenizer = processor.tokenizer
         
         logger.info("Loading model...")
-        # Use AutoModelForCausalLM for better compatibility
-        model = AutoModelForCausalLM.from_pretrained(
+        # Use specific Qwen2VL model class
+        model = Qwen2VLForConditionalGeneration.from_pretrained(
             model_name,
             torch_dtype=torch.float32,
             device_map=None,  # CPU only
@@ -248,12 +248,4 @@ async def health_check():
         "device": "cpu",
         "torch_dtype": "float32",
         "model_loaded": model_loaded
-    }
-
-@app.post("/click/form")
-async def predict_click_form(
-    image_base64: str = Form(...),
-    instruction: str = Form(...)
-):
-    data = Base64Request(image_base64=image_base64, instruction=instruction)
-    return await predict_click_base64(data)
+    }