addlogin

app.py

@@ -4,6 +4,11 @@ import numpy as np
 import random
 import torch
 from diffusers import StableDiffusion3Pipeline
+from huggingface_hub import login
+import os
+
+# Login to Hugging Face using environment variable
+login(token=os.getenv("HF_TOKEN"))
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32
@@ -131,4 +136,4 @@ with gr.Blocks(css=css) as demo:
         outputs=[result, seed]
     )
 
-demo.launch()
+demo.launch()