Update app.py

app.py

@@ -6,15 +6,15 @@ import matplotlib.pyplot as plt
 from transformers import MarianMTModel, MarianTokenizer, pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
+import os  # For accessing environment variables
 
 # Constants for model names and API URLs
 class Constants:
     TRANSLATION_MODEL_NAME = "Helsinki-NLP/opus-mt-mul-en"
     IMAGE_GENERATION_API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
     GPT_NEO_MODEL_NAME = "EleutherAI/gpt-neo-125M"
-    HEADERS = {"Authorization": "Bearer {os.getenv('hf_token')}"}
-    
-
+    # Get the Hugging Face API token from environment variables
+    HEADERS = {"Authorization": f"Bearer {os.getenv('HF_API_TOKEN')}"}
 
 # Translation Class
 class Translator: