Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,6 @@ def predict(img, text):
|
|
| 11 |
img,
|
| 12 |
text
|
| 13 |
)
|
| 14 |
-
|
| 15 |
-
return prediction['answer']
|
| 16 |
|
| 17 |
gr.Interface(predict, [gr.Image(type="filepath"), gr.Textbox(label="Question")], gr.Textbox()).launch()
|
|
|
|
| 11 |
img,
|
| 12 |
text
|
| 13 |
)
|
| 14 |
+
return prediction[0]['answer']
|
|
|
|
| 15 |
|
| 16 |
gr.Interface(predict, [gr.Image(type="filepath"), gr.Textbox(label="Question")], gr.Textbox()).launch()
|