Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|