Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
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 |
|