Spaces:

oshita-n
/

ImageQuestionAnswerring

Runtime error

oshita-n commited on Mar 25, 2023

Commit

577ce6a

1 Parent(s): 0f3dfdc

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def process(input_image, prompt):
     text_input = txt_processors["eval"](prompt)
     sample = {"image": image, "text_input": [text_input]}
-    output = model.forward_qa(samples=sample)
     pred_answers = output[0]
     return pred_answer

     text_input = txt_processors["eval"](prompt)
     sample = {"image": image, "text_input": [text_input]}
+    output = model.forward_qa(samples=sample, num_captions=50)
     pred_answers = output[0]
     return pred_answer