ManishThota commited on
Commit
2c37e20
1 Parent(s): fcb3375

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
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 ast.literal_eval(answers[0])
102
 
103
  else:
104
  return "Unsupported file type. Please upload an image or video."