akhaliq HF staff commited on
Commit
4ed47d4
1 Parent(s): 2f426cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -172,14 +172,14 @@ def inference(img):
172
 
173
  sents = utils.decode_sequence(model.vocab, seq)
174
 
175
- return sents
176
 
177
  demo = gr.Blocks()
178
 
179
  with demo:
180
  gr.Markdown(
181
  """
182
- # Demo for CLIP-Caption-Reward
183
  """)
184
  inp = gr.Image(type="pil")
185
  out = gr.Textbox()
172
 
173
  sents = utils.decode_sequence(model.vocab, seq)
174
 
175
+ return sents[0]
176
 
177
  demo = gr.Blocks()
178
 
179
  with demo:
180
  gr.Markdown(
181
  """
182
+ # Gradio Demo for [j-min/CLIP-Caption-Reward](https://github.com/j-min/CLIP-Caption-Reward)
183
  """)
184
  inp = gr.Image(type="pil")
185
  out = gr.Textbox()