Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel 
 
 # Load text generation model
-# text_generation_model = pipeline("text-generation", model="openai-community/gpt2")
-text_generation_model = pipeline("text-generation", model="distilbert/distilgpt2")
+text_generation_model = pipeline("text-generation", model="openai-community/gpt2-large")
+# text_generation_model = pipeline("text-generation", model="distilbert/distilgpt2")
 
 # Load image captioning model
 encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"