Update app.py

app.py

@@ -6,6 +6,11 @@ import io
 from PIL import Image
 import os
 
+# Load the Hugging Face API key from environment variables
+hf_api_key = os.getenv("full_token")
+if hf_api_key is None:
+    raise ValueError("Hugging Face API key not found! Please set the 'full_token' environment variable.")
+
 # Load the translation model and tokenizer
 model_name = "facebook/mbart-large-50-many-to-one-mmt"
 tokenizer = MBart50Tokenizer.from_pretrained(model_name)
@@ -18,8 +23,8 @@ text_model = AutoModelForCausalLM.from_pretrained(text_generation_model_name)
 text_generator = pipeline("text-generation", model=text_model, tokenizer=text_tokenizer)
 
 # Load the FLUX.1-dev image generation model from diffusers
-flux_model = StableDiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
-flux_model.to("cuda")  # Make sure to use a GPU for faster generation
+flux_model = StableDiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", use_auth_token=hf_api_key)
+flux_model.to("cuda")  # Use GPU for faster generation, if available
 
 # Function to generate an image using FLUX.1-dev model
 def generate_image_from_text(translated_text):