Update app.py

app.py

@@ -7,8 +7,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_write"
+  model_ckpt = "bankholdup/rugpt3_song_writer"
+#  model_ckpt = "bankholdup/mgpt_song_write"
   tokenizer = GPT2Tokenizer.from_pretrained(model_ckpt)
   model = GPT2LMHeadModel.from_pretrained(model_ckpt)
   return tokenizer, model