Spaces:
Runtime error
Runtime error
fix
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ def process(input_image, prompt):
|
|
| 13 |
sample = {"image": image, "text_input": [text_input]}
|
| 14 |
pred_answers, caption, gradcam = model.predict_answers(sample, num_captions=50, num_patches=20)
|
| 15 |
|
| 16 |
-
return
|
| 17 |
|
| 18 |
if __name__ == '__main__':
|
| 19 |
input_image = gr.inputs.Image(label='image', type='pil')
|
|
|
|
| 13 |
sample = {"image": image, "text_input": [text_input]}
|
| 14 |
pred_answers, caption, gradcam = model.predict_answers(sample, num_captions=50, num_patches=20)
|
| 15 |
|
| 16 |
+
return pred_answers[0]
|
| 17 |
|
| 18 |
if __name__ == '__main__':
|
| 19 |
input_image = gr.inputs.Image(label='image', type='pil')
|