Update app.py

app.py

@@ -1,3 +1,5 @@
+#!/usr/bin/env python
+#patch 0.01
 import os
 import random
 import uuid
@@ -137,12 +139,15 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
+huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
+
 if torch.cuda.is_available():
     repo = "stabilityai/stable-diffusion-3-medium"  # You might want to replace with your specific repo if different
     pipe = StableDiffusion3Pipeline.from_pretrained(
         repo,
         torch_dtype=torch.float16,
-        revision="refs/pr/26"
+        revision="refs/pr/26",
+        use_auth_token=huggingface_token
     ).to(device)
 
     if ENABLE_CPU_OFFLOAD:
@@ -409,4 +414,4 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
         generator = generator
     ).images[0] 
     
-    return image, seed
+    return image, seed