Spaces:

ubermenchh
/

caption_it

Runtime error

ubermenchh commited on Nov 29, 2023

Commit

6fa39c4

•

1 Parent(s): b862c75

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,8 +21,8 @@ vis_config = Blip2VisionConfig()
 model = Blip2VisionModel(vis_config)
 config_2 = model.config
-processor = AutoProcessor.from_pretrained('Salesforce/blip2-opt-2.7b')
-model = Blip2ForConditionalGeneration.from_pretrained('Salesforce/blip2-opt-2.7b')
 def captioning(image):
     inputs = processor(images=image, return_tensors='pt')

 model = Blip2VisionModel(vis_config)
 config_2 = model.config
+processor = AutoProcessor.from_pretrained('Salesforce/blip-image-captioning-large')
+model = Blip2ForConditionalGeneration.from_pretrained('Salesforce/blip-image-captioning-large')
 def captioning(image):
     inputs = processor(images=image, return_tensors='pt')