Update name of model

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-from transformers import ViltProcessor, ViltForVisualQuestionAnswering
+from transformers import ViltProcessor, ViltForQuestionAnswering
 import torch
 
 torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/000000039769.jpg', 'cats.jpg')
 torch.hub.download_url_to_file('https://computing.ece.vt.edu/~harsh/visualAttention/ProjectWebpage/Figures/vqa_1.png', 'banana.png')
 
 processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
-model = ViltForVisualQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
+model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 
 def answer_question(image, text):
     encoding = processor(image, text, return_tensors="pt")