wahdia commited on
Commit
ef357c3
·
verified ·
1 Parent(s): a51fd05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -4,13 +4,18 @@ import torch
4
  from transformers import pipeline
5
  from datetime import datetime
6
 
7
- from huggingface_hub import login
8
  import os
 
9
 
 
 
 
 
10
 
 
 
11
 
12
- # Option 2 (RECOMMANDÉE) : utilisez une variable d'environnement
13
- login(token=os.getenv("hf_token"))
14
 
15
 
16
 
 
4
  from transformers import pipeline
5
  from datetime import datetime
6
 
 
7
  import os
8
+ from huggingface_hub import login
9
 
10
+ # Get token from environment variable
11
+ hf_token = os.getenv("hf_token")
12
+ if hf_token is None:
13
+ raise ValueError("Please set HF_TOKEN environment variable with your Hugging Face token")
14
 
15
+ # Login with the token
16
+ login(token=hf_token)
17
 
18
+ # Rest of your code...
 
19
 
20
 
21