Hugging Face's logo

Spaces:
m7mdal7aj
/
KB-VQA
Sleeping

App Files Community
2
m7mdal7aj commited on
Commit
b9be639
·
1 Parent(s): bc7d231

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from transformers import Blip2Processor, Blip2ForConditionalGeneration
5
 
6
 
7
- def load_model():
8
 
9
  processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
10
  model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16, device_map="auto")
 
4
  from transformers import Blip2Processor, Blip2ForConditionalGeneration
5
 
6
 
7
+ def load_caption_model():
8
 
9
  processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
10
  model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16, device_map="auto")