Hugging Face's logo

Spaces:
skylersterling
/
TopicGPT
Sleeping

App Files Community
skylersterling commited on
Commit
5e2cdce
·
verified ·
1 Parent(s): f6a7df4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
  # Load the tokenizer and model
10
- tokenizer = GPT2Tokenizer.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
11
  model = GPT2LMHeadModel.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
12
  model.eval()
13
 
 
7
  HF_TOKEN = os.environ.get("HF_TOKEN")
8
 
9
  # Load the tokenizer and model
10
+ tokenizer = GPT2Tokenizer.from_pretrained('gpt2', use_auth_token=HF_TOKEN)
11
  model = GPT2LMHeadModel.from_pretrained('skylersterling/TopicGPT', use_auth_token=HF_TOKEN)
12
  model.eval()
13