Spaces:

gopalagra
/

blind-image-captioning

Running

gopalagra commited on Sep 7

Commit

d11dc78

verified ·

1 Parent(s): 6ec7c3f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -74,15 +74,15 @@ import torch
 # ----------------------
 # Load BLIP2 for Captioning
 # ----------------------
-caption_processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
-caption_model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
 # ----------------------
 # Load BLIP2 for VQA
 # ----------------------
-vqa_processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-xl")
 vqa_model = Blip2ForConditionalGeneration.from_pretrained(
-    "Salesforce/blip2-flan-t5-xl", torch_dtype=torch.float16, device_map="auto"
 )
 # ----------------------

 # ----------------------
 # Load BLIP2 for Captioning
 # ----------------------
+caption_processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-125m")
+caption_model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-125m")
 # ----------------------
 # Load BLIP2 for VQA
 # ----------------------
+vqa_processor = Blip2Processor.from_pretrained("Salesforce/blip2-flan-t5-base")
 vqa_model = Blip2ForConditionalGeneration.from_pretrained(
+   "Salesforce/blip2-flan-t5-base", torch_dtype=torch.float16, device_map="auto"
 )
 # ----------------------