jaimin commited on
Commit
2b3702d
·
1 Parent(s): 0b66865

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -30,11 +30,11 @@ def generate_caption(processor, model, image, use_float_16=False):
30
 
31
  def generate_captions(image):
32
 
33
- img = Image.open(image)
34
- caption_git = generate_caption(git_processor, git_model, img)
35
 
36
- caption_blip = generate_caption(blip_processor, blip_model, img)
37
- env, scene = predict_environment(img)
38
 
39
  return env,scene,caption_git_large_textcaps, caption_blip_large
40
 
 
30
 
31
  def generate_captions(image):
32
 
33
+ #img = Image.open(image)
34
+ caption_git = generate_caption(git_processor, git_model, image)
35
 
36
+ caption_blip = generate_caption(blip_processor, blip_model, image)
37
+ env, scene = predict_environment(image)
38
 
39
  return env,scene,caption_git_large_textcaps, caption_blip_large
40