Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,10 +10,6 @@ from transformers import Blip2Processor, Blip2ForConditionalGeneration, Instruct
|
|
10 |
|
11 |
def load_caption_model(blip2=False, instructblip=True):
|
12 |
|
13 |
-
|
14 |
-
model = YourModel()
|
15 |
-
|
16 |
-
|
17 |
if blip2:
|
18 |
processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
19 |
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
|
|
10 |
|
11 |
def load_caption_model(blip2=False, instructblip=True):
|
12 |
|
|
|
|
|
|
|
|
|
13 |
if blip2:
|
14 |
processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|
15 |
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b", load_in_8bit=True,torch_dtype=torch.float16)
|