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')
|