Spaces:
Paused
Paused
Update app.py
Browse files
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
|
| 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."
|