elapt1c commited on
Commit
a01bbdc
·
verified ·
1 Parent(s): 6292c12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,8 +41,8 @@ def generate_response(input_text):
41
  return bot_response
42
 
43
  # Load pre-trained model tokenizer (vocabulary) and model
44
- model_name = 'KhantKyaw/Chat_GPT-2'
45
- tokenizer = GPT2Tokenizer.from_pretrained(model_name)
46
  model = GPT2LMHeadModel.from_pretrained(model_name)
47
 
48
  st.title("Chat_GPT-2 Bot")
 
41
  return bot_response
42
 
43
  # Load pre-trained model tokenizer (vocabulary) and model
44
+ model_name = 'elapt1c/ElapticAI-1a'
45
+ tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
46
  model = GPT2LMHeadModel.from_pretrained(model_name)
47
 
48
  st.title("Chat_GPT-2 Bot")