Update app.py

app.py

@@ -12,7 +12,7 @@ def load_model(model_name):
     return model
     
 def load_text_gen_model():
-    generator = pipeline("text-generation", model="gpt2-medium")
+    generator = pipeline("text-generation", model="gpt2-large")
     return generator 
     
 model = load_model("gpt2-large")