Update app.py

app.py

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