ManishThota commited on
Commit
fcb3375
1 Parent(s): 89af6b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def predict_answer(video, image, question):
98
 
99
  answer = tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip()
100
  answers.append(answer)
101
- return answers[1]
102
 
103
  else:
104
  return "Unsupported file type. Please upload an image or video."
 
98
 
99
  answer = tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip()
100
  answers.append(answer)
101
+ return answers
102
 
103
  else:
104
  return "Unsupported file type. Please upload an image or video."