gxy commited on
Commit
a08f8d9
1 Parent(s): df08d3f

Update app.py

Browse files

add test print

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ def inference(raw_image, model_n, strategy):
35
  captions = model.generate(**input, generation_config=config)
36
  caption = processor.decode(captions[0], skip_special_tokens=True)
37
  caption = caption.replace(' ', '')
 
38
  return 'caption: '+caption
39
 
40
 
 
35
  captions = model.generate(**input, generation_config=config)
36
  caption = processor.decode(captions[0], skip_special_tokens=True)
37
  caption = caption.replace(' ', '')
38
+ print(caption)
39
  return 'caption: '+caption
40
 
41