Update app.py

app.py

@@ -81,8 +81,9 @@ 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("EleutherAI/gpt-neo-1.3B")
-gpt_model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
+gpt_tokenizer = AutoTokenizer.from_pretrained("distilgpt2")
+gpt_model = AutoModelForCausalLM.from_pretrained("distilgpt2")
+
 
 def rewrite_answer(question):
     prompt = f"{question}\nAnswer with a full sentence:"