oshita-n commited on
Commit
7a1a0e9
·
1 Parent(s): 9d6822a
Files changed (1) hide show
  1. app.py +1 -1
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 pred_answer
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')