aronvandepol commited on
Commit
af05640
·
1 Parent(s): 34ea598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
5
 
6
  auth_token = os.environ.get("TOKEN") or True
7
 
8
- model = AutoModelForCausalLM.from_pretrained("aronvandepol/K-GPT125M", use_auth_token=)
9
 
10
  tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
11
 
 
5
 
6
  auth_token = os.environ.get("TOKEN") or True
7
 
8
+ model = AutoModelForCausalLM.from_pretrained("aronvandepol/K-GPT125M", use_auth_token=auth_token)
9
 
10
  tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neo-125M")
11