Update image_pipeline.py

image_pipeline.py

@@ -1,10 +1,10 @@
-from transformers import AutoProcessor, AutoModelForImageTextToText
+from transformers import AutoProcessor, AutoModelForVision2Seq
 from PIL import Image
 from config import HF_IMAGE_MODEL
 
 # Load Hugging Face image model and processor
 processor = AutoProcessor.from_pretrained(HF_IMAGE_MODEL)
-model = AutoModelForImageTextToText.from_pretrained(HF_IMAGE_MODEL)
+model = AutoModelForVision2Seq.from_pretrained(HF_IMAGE_MODEL)
 
 def analyze_medical_image(image_file):
     """