Update app.py

app.py

@@ -8,10 +8,17 @@ from huggingface_hub import login
 hf_token = os.getenv("HF_TOKEN")  # This should be set in your Hugging Face Space Secrets
 login(token=hf_token)
 
+# Access the NVIDIA API key from the environment
+nvidia_api_key = os.getenv("NVIDIA_API_KEY")
+
 # Load the model using the API key and move it to GPU
 def load_model():
     model_id = "NVIDIA/sdxl-turbo"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16, use_auth_token=True)
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id, 
+        torch_dtype=torch.float16, 
+        use_auth_token=nvidia_api_key  # Use the API key here
+    )
     pipe = pipe.to("cuda")
     return pipe