Update app.py

app.py

@@ -1,9 +1,14 @@
+import os
 import gradio as gr
 import torch
 from diffusers import StableDiffusion3Pipeline
 
 def image_generation(prompt):
     device = "cuda" if torch.cuda.is_available() else "cpu"
+    # Retrieve the token from the environment variable
+    huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+    print(os.getenv("HUGGINGFACE_TOKEN"))
+
     pipeline = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", 
                                                         torch_dtype=torch.float16 if device == "cuda" else torch.float32,
                                                         text_encoder_3=None,