Update app.py

app.py

@@ -5,10 +5,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Load environment variables
-load_dotenv()
-HF_TOKEN = os.getenv("HF_TOKEN")
 
+load_dotenv()
 # Set Streamlit page config
 st.set_page_config(
     page_title="Tamil Creative Studio",
@@ -24,29 +22,31 @@ def load_css(file_name):
 
 @st.cache_resource(show_spinner=False)
 def load_all_models():
+    HF_TOKEN = os.getenv("HF_TOKEN")  # Hugging Face token from Hugging Face Spaces secret
+
     # Load translation model (private)
     trans_tokenizer = AutoTokenizer.from_pretrained(
         "ai4bharat/indictrans2-ta-en-dist-200M",
-        use_auth_token=HF_TOKEN
+        token=HF_TOKEN
     )
     trans_model = AutoModelForSeq2SeqLM.from_pretrained(
         "ai4bharat/indictrans2-ta-en-dist-200M",
-        use_auth_token=HF_TOKEN
+        token=HF_TOKEN
     )
-    
+
     # Load text generation model
     text_gen = pipeline("text-generation", model="gpt2", device=-1)
-    
+
     # Load image generation model
     img_pipe = StableDiffusionPipeline.from_pretrained(
         "stabilityai/stable-diffusion-2-base",
-        use_auth_token=HF_TOKEN,
         torch_dtype=torch.float32,
         safety_checker=None
     ).to("cpu")
-    
+
     return trans_tokenizer, trans_model, text_gen, img_pipe
 
+
 def translate_tamil(text, tokenizer, model):
     inputs = tokenizer(
         text,