imalexianne commited on
Commit
50b8f28
1 Parent(s): 5887865

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -8,16 +8,12 @@ from huggingface_hub import login
8
  from dotenv import load_dotenv
9
  load_dotenv()
10
 
11
- # Get the token from the environment variable
12
- access_token = os.getenv("access_token")
13
-
14
- # Log in to Hugging Face (commented out for now)
15
- # login(access_token)
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
 
 
8
  from dotenv import load_dotenv
9
  load_dotenv()
10
 
11
+ login(os.getenv("access_token"))
 
 
 
 
12
 
13
  # Requirements
14
  model_path = "imalexianne/distilbert-base-uncased"
15
  # tokenizer = AutoTokenizer.from_pretrained(model_path)
16
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
17
 
18
  model = AutoModel.from_pretrained(model_path)
19