pravin007s commited on
Commit
0791989
·
verified ·
1 Parent(s): 0ff1b8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -7,9 +7,19 @@ Original file is located at
7
  https://colab.research.google.com/drive/1iF7hdOjWNeFUtGvUYdaFsBErJGnY1h5J
8
  """
9
 
 
10
  from huggingface_hub import login
11
 
12
- login(token="HF_TOKEN")
 
 
 
 
 
 
 
 
 
13
 
14
  # Import necessary libraries
15
  from transformers import MarianMTModel, MarianTokenizer, pipeline
 
7
  https://colab.research.google.com/drive/1iF7hdOjWNeFUtGvUYdaFsBErJGnY1h5J
8
  """
9
 
10
+ import os
11
  from huggingface_hub import login
12
 
13
+ # Retrieve the actual token from the environment variable
14
+ hf_token = os.getenv("HF_TOKEN")
15
+
16
+ # Check if the token is retrieved properly
17
+ if hf_token:
18
+ # Use the retrieved token
19
+ login(token=hf_token, add_to_git_credential=True)
20
+ else:
21
+ raise ValueError("Hugging Face token not found in environment variables.")
22
+
23
 
24
  # Import necessary libraries
25
  from transformers import MarianMTModel, MarianTokenizer, pipeline