ManishThota commited on
Commit
f28f6f6
1 Parent(s): 5c72980

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(image, video, question):
98
 
99
  answer = tokenizer.decode(output_ids[input_ids.shape[1]:], skip_special_tokens=True).strip()
100
  answers.append(answer)
101
- return "\n".join(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."