Update app.py

app.py

@@ -2,8 +2,8 @@ import streamlit as st
 
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 model_name = 'elapt1c/ElapticAI-1a'
-tokenizer = GPT2Tokenizer.from_pretrained(model_name)
-model = GPT2LMHeadModel.from_pretrained("gpt2")
+tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
+model = GPT2LMHeadModel.from_pretrained(model_name)
 
 special_tokens_dict = {'bos_token': '<BOS>', 'eos_token': '<EOS>', 'sep_token': '<SEP>', 'pad_token': '<PAD>'}
 tokenizer.add_special_tokens(special_tokens_dict)