Spaces:
Runtime error
Runtime error
Commit
·
3c04d54
1
Parent(s):
1b1e4db
Update app.py
Browse files
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 |
-
|
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,
|
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"]]
|