k

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from PIL import Image
 import torch
-from transformers import Blip2Processor, Blip2ForConditionalGeneration
+from transformers import BlipProcessor, BlipForQuestionAnswering
 
-# Load the processor and model
-processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
-model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", torch_dtype=torch.float16)
+# Load processor and small BLIP VQA model
+processor = BlipProcessor.from_pretrained("Salesforce/blip-vqa-base")
+model = BlipForQuestionAnswering.from_pretrained("Salesforce/blip-vqa-base")
 
 # Use CPU explicitly
 device = torch.device("cpu")