Update app.py

app.py

@@ -4,7 +4,7 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelWithLMHead
 from transformers import pipeline
 
-#tokenizer = AutoTokenizer.from_pretrained("gpt2-small")
+#tokenizer = AutoTokenizer.from_pretrained("gpt2-medium")
 
 @st.cache
 def load_model(model_name):
@@ -12,10 +12,10 @@ def load_model(model_name):
     return model
     
 def load_text_gen_model():
-    generator = pipeline("text-generation", model="gpt2-small")
+    generator = pipeline("text-generation", model="gpt2-medium")
     return generator 
     
-#model = load_model("gpt2-small")
+#model = load_model("gpt2-medium")
 
 text_generator = load_text_gen_model()