Update app.py

app.py

@@ -1,10 +1,19 @@
+import os
 import gradio as gr
 import numpy as np
 import random
-
-# import spaces #[uncomment to use ZeroGPU]
+import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
 import torch
+from huggingface_hub import login
+
+
+HF_TOKEN = os.getenv('HF_TOKEN')
+
+if not HF_TOKEN:
+    raise ValueError("Please set your Hugging Face API token in a .env file")
+
+login(token=HF_TOKEN)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"