Update app.py

app.py

@@ -3,6 +3,7 @@ import requests
 from transformers import MarianMTModel, MarianTokenizer, AutoModelForCausalLM, AutoTokenizer
 from PIL import Image
 import torch
+import os
 import io
 
 # Check if GPU is available
@@ -21,9 +22,8 @@ if generator_tokenizer.pad_token is None:
     generator_tokenizer.pad_token = generator_tokenizer.eos_token
 
 # Hugging Face API for Image Generation
-HF_API_KEY = os.getenv("HF_API_KEY")  # Replace with your API key
+HF_API_KEY = os.getenv("HF_API_KEY")  # Use environment variable
 IMAGE_GEN_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
-# Get the API key from environment variables or Hugging Face secrets
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 def translate_tamil_to_english(text):