Hugging Face's logo

Spaces:
ceyda
/
comparing-captioning-models
Runtime error

App Files Community
nielsr HF staff commited on
Commit
ef34b5f
·
1 Parent(s): 3072768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def generate_captions(image):
85
 
86
  caption_coca = generate_caption_coca(coca_model, coca_transform, image)
87
 
88
- caption_blip2 = generate_caption(blip2_processor, blip2_model, image, use_float_16=True)
89
 
90
  return caption_git_large_coco, caption_git_large_textcaps, caption_blip_large, caption_coca, caption_blip2
91
 
 
85
 
86
  caption_coca = generate_caption_coca(coca_model, coca_transform, image)
87
 
88
+ caption_blip2 = generate_caption(blip2_processor, blip2_model, image, use_float_16=True).strip()
89
 
90
  return caption_git_large_coco, caption_git_large_textcaps, caption_blip_large, caption_coca, caption_blip2
91