Update app.py

app.py

@@ -8,16 +8,18 @@ from diffusers import DiffusionPipeline
 import torch
 from huggingface_hub import login
 
-import yaml
-
-with open("config.yaml", "r") as f:
-    config = yaml.safe_load(f)
+import os
+from huggingface_hub import login
 
-token = config.get("huggingface_token")
+# Access the token from environment variables
+token = os.getenv("HUGGINGFACE_API_TOKEN")
+if not token:
+    raise ValueError("API token is missing! Please set it in the Hugging Face Space Secrets.")
 
-# Login to Hugging Face Hub
+# Log in to Hugging Face
 login(token)
 
+
 # Model details
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32