Update app.py

app.py

@@ -9,9 +9,16 @@ from peft import PeftModel
 import os
 from huggingface_hub import snapshot_download
 
-auth_token = os.getenv('HUGGINFACE_TOKEN')
-
-model_path = "stabilityai/stable-diffusion-3-medium-diffusers"
+huggingface_token = os.getenv("HUGGINFACE_TOKEN")
+
+model_path = snapshot_download(
+    repo_id="stabilityai/stable-diffusion-3-medium", 
+    revision="refs/pr/26",
+    repo_type="model", 
+    ignore_patterns=["*.md", "*..gitattributes"],
+    local_dir="stable-diffusion-3-medium",
+    token=huggingface_token, # type a new token-id.
+    )
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 IS_SPACE = os.environ.get("SPACE_ID", None) is not None