m7mdal7aj commited on
Commit
8a94bf7
1 Parent(s): 63fc765

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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)