imalexianne commited on
Commit
defc0a9
1 Parent(s): 8a02fed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -16,7 +16,9 @@ access_token = os.getenv("access_token")
16
 
17
  # Requirements
18
  model_path = "imalexianne/distilbert-base-uncased"
19
- tokenizer = AutoTokenizer.from_pretrained(model_path)
 
 
20
  model = AutoModel.from_pretrained(model_path)
21
 
22
  # Preprocessing function
 
16
 
17
  # Requirements
18
  model_path = "imalexianne/distilbert-base-uncased"
19
+ # tokenizer = AutoTokenizer.from_pretrained(model_path)
20
+ tokenizer = AutoTokenizer.from_pretrained(model_path, revision="main", use_auth_token=os.getenv("access_token"))
21
+
22
  model = AutoModel.from_pretrained(model_path)
23
 
24
  # Preprocessing function