Update app.py

app.py

@@ -8,10 +8,8 @@ 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"
   tokenizer = GPT2Tokenizer.from_pretrained(model_ckpt)
   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):