elapt1c commited on
Commit
46ce815
·
verified ·
1 Parent(s): dc31ddc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def generate_response(input_text):
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
 
 
41
  return bot_response
42
 
43
  # Load pre-trained model tokenizer (vocabulary) and model
44
+ model_name = 'gpt2'
45
  tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
46
  model = GPT2LMHeadModel.from_pretrained(model_name)
47