Update app.py

app.py

@@ -81,8 +81,10 @@ vqa_processor = ViltProcessor.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 vqa_model = ViltForQuestionAnswering.from_pretrained("dandelin/vilt-b32-finetuned-vqa")
 
 # Load GPT model to rewrite answers
-gpt_tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
-gpt_model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+
+gpt_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-small")
+gpt_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-small")
+
 
 def rewrite_answer(question, short_answer):
     prompt = (