yakine commited on
Commit
6934761
·
verified ·
1 Parent(s): 920ca00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,14 +5,14 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer, AutoTokenizer, pipeline
5
  from huggingface_hub import HfFolder
6
  from io import StringIO
7
  import os
8
- huggingface-cli login
9
 
10
  # Access the Hugging Face API token from environment variables
11
  hf_token = os.getenv('HF_API_TOKEN')
12
 
13
  if not hf_token:
14
  raise ValueError("Hugging Face API token is not set. Please set the HF_API_TOKEN environment variable.")
15
-
16
  # Set environment variable to avoid floating-point errors
17
  os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
18
 
 
5
  from huggingface_hub import HfFolder
6
  from io import StringIO
7
  import os
8
+
9
 
10
  # Access the Hugging Face API token from environment variables
11
  hf_token = os.getenv('HF_API_TOKEN')
12
 
13
  if not hf_token:
14
  raise ValueError("Hugging Face API token is not set. Please set the HF_API_TOKEN environment variable.")
15
+ HfFolder.save_token(hf_token)
16
  # Set environment variable to avoid floating-point errors
17
  os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
18