Update app.py

app.py

@@ -13,7 +13,7 @@ def load_model(model_name):
     
 @st.cache
 def load_text_gen_model():
-    generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+    generator = pipeline("text-generation", model="gpt2-medium")
     return generator 
     
 model = load_model("gpt2-large")