ubermenchh commited on
Commit
6fa39c4
1 Parent(s): b862c75

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -21,8 +21,8 @@ vis_config = Blip2VisionConfig()
21
  model = Blip2VisionModel(vis_config)
22
  config_2 = model.config
23
 
24
- processor = AutoProcessor.from_pretrained('Salesforce/blip2-opt-2.7b')
25
- model = Blip2ForConditionalGeneration.from_pretrained('Salesforce/blip2-opt-2.7b')
26
 
27
  def captioning(image):
28
  inputs = processor(images=image, return_tensors='pt')
 
21
  model = Blip2VisionModel(vis_config)
22
  config_2 = model.config
23
 
24
+ processor = AutoProcessor.from_pretrained('Salesforce/blip-image-captioning-large')
25
+ model = Blip2ForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-large')
26
 
27
  def captioning(image):
28
  inputs = processor(images=image, return_tensors='pt')