req changed

model.py

@@ -1,14 +1,13 @@
 import torch
-from transformers import BlipProcessor, Blip2ForConditionalGeneration
+from transformers import AutoProcessor, AutoModelForVisualQuestionAnswering
 from PIL import Image
 
 # Load BLIP-2 model and processor
-processor = BlipProcessor.from_pretrained("Salesforce/blip2-flan-t5-xl")
-model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-flan-t5-xl")
+# Load model directly
+processor = AutoProcessor.from_pretrained("Salesforce/blip2-flan-t5-xl")
+model = AutoModelForVisualQuestionAnswering.from_pretrained("Salesforce/blip2-flan-t5-xl")
 
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-
 def generate_answer(image: Image.Image, question: str) -> str:
     if not image.mode == "RGB":
         image = image.convert("RGB")