Hugging Face's logo

Spaces:
miculpionier
/
Visual-Question-Answering
Runtime error

App Files Community
miculpionier commited on
Commit
66e49d4
•
1 Parent(s): 66b4c6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio
2
  from transformers import ViltProcessor, ViltForQuestionAnswering
3
  from PIL import Image
4
 
5
- processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
6
- model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
7
 
8
 
9
  def predict_answer(image, question):
 
2
  from transformers import ViltProcessor, ViltForQuestionAnswering
3
  from PIL import Image
4
 
5
+ processor = ViltProcessor.from_pretrained("vilt-b32-finetuned-vqa")
6
+ model = ViltForQuestionAnswering.from_pretrained("vilt-b32-finetuned-vqa")
7
 
8
 
9
  def predict_answer(image, question):