Update app.py

app.py

@@ -1,5 +1,14 @@
 from diffusers import StableDiffusionPipeline
 import torch
+import os
+from huggingface_hub import login
+
+# Retrieve the token from the environment variable
+token = os.getenv("HF_TOKEN")  # Hugging Face token from the secret
+if token:
+    login(token=token)  # Log in with the retrieved token
+else:
+    raise ValueError("Hugging Face token not found. Please set it as a repository secret in the Space settings.")
 
 # Load the Stable Diffusion 3.5 model
 model_id = "stabilityai/stable-diffusion-3.5"