pravin0077 commited on
Commit
aba5af4
·
verified ·
1 Parent(s): 630f14f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ class Constants:
14
  IMAGE_GENERATION_API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
15
  GPT_NEO_MODEL_NAME = "EleutherAI/gpt-neo-125M"
16
  # Get the Hugging Face API token from environment variables
17
- HEADERS = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
18
 
19
  # Translation Class
20
  class Translator:
 
14
  IMAGE_GENERATION_API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
15
  GPT_NEO_MODEL_NAME = "EleutherAI/gpt-neo-125M"
16
  # Get the Hugging Face API token from environment variables
17
+ HEADERS = {"Authorization": f"Bearer {os.getenv('HUGGINGFACE_API_KEY')}"}
18
 
19
  # Translation Class
20
  class Translator: