Update app.py

app.py

@@ -7,11 +7,11 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
 
 @st.cache(allow_output_mutation=True)
 def load_model():
-#  model_ckpt = "bankholdup/rugpt3_song_writer"
-  model_ckpt = "bankholdup/mgpt_song_writer"
+  model_ckpt = "bankholdup/rugpt3_song_writer"
+#  model_ckpt = "bankholdup/mgpt_song_writer"
   tokenizer = GPT2Tokenizer.from_pretrained(model_ckpt)
-#  model = GPT2LMHeadModel.from_pretrained(model_ckpt)
-  model = GPT2LMHeadModel.from_pretrained(model_ckpt, low_cpu_mem_usage=True)
+  model = GPT2LMHeadModel.from_pretrained(model_ckpt)
+#  model = GPT2LMHeadModel.from_pretrained(model_ckpt, low_cpu_mem_usage=True)
   return tokenizer, model
 
 def set_seed(rng=100000):