wahdia commited on
Commit
cbd6c9b
·
verified ·
1 Parent(s): 4c58c32

Update app (1).py

Browse files
Files changed (1) hide show
  1. app (1).py +10 -0
app (1).py CHANGED
@@ -4,6 +4,16 @@ import transformers
4
  import torch
5
  from PIL import Image
6
 
 
 
 
 
 
 
 
 
 
 
7
  # Chargement des modèles
8
  translator = transformers.pipeline("translation", model="facebook/nllb-200-distilled-600M")
9
  text_gen_pipeline = transformers.pipeline(
 
4
  import torch
5
  from PIL import Image
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
  # Chargement des modèles
18
  translator = transformers.pipeline("translation", model="facebook/nllb-200-distilled-600M")
19
  text_gen_pipeline = transformers.pipeline(