multimodalart HF staff commited on
Commit
3c04d54
·
1 Parent(s): 1b1e4db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,11 +58,11 @@ def generate_captions(image):
58
 
59
  caption_blip_large = generate_caption(blip_processor_large, blip_model_large, image)
60
 
61
- caption_blip2_8_bit = generate_caption_blip2(blip2_processor, blip2_model_8_bit, image).strip()
62
 
63
  caption_instructblip_4_bit = generate_caption_blip2(instructblip_processor, instructblip_model_4_bit, image)
64
 
65
- return caption_git_large_coco, caption_blip_large, caption_blip2_8_bit, caption_instructblip_4_bit
66
 
67
 
68
  examples = [["cats.jpg"], ["stop_sign.png"], ["astronaut.jpg"]]
 
58
 
59
  caption_blip_large = generate_caption(blip_processor_large, blip_model_large, image)
60
 
61
+ caption_blip2_4_bit = generate_caption_blip2(blip2_processor, blip2_model_4_bit, image).strip()
62
 
63
  caption_instructblip_4_bit = generate_caption_blip2(instructblip_processor, instructblip_model_4_bit, image)
64
 
65
+ return caption_git_large_coco, caption_blip_large, caption_blip2_4_bit, caption_instructblip_4_bit
66
 
67
 
68
  examples = [["cats.jpg"], ["stop_sign.png"], ["astronaut.jpg"]]