Update app.py

app.py

@@ -8,9 +8,20 @@ from huggingface_hub import hf_hub_download
 import torch 
 import os
 from PIL import Image, ImageFilter, ImageOps
+from huggingface_hub import login, hf_hub_download
+import os
+
+# Log in to Hugging Face for private repo access
+if "HF_TOKEN" in os.environ:
+    login(os.environ["HF_TOKEN"])
+else:
+    raise ValueError("HF_TOKEN not found in environment variables. Please set it in Space settings.")
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model_path = "https://huggingface.co/DreamingOracle/Quagmaform_alpha-1/blob/main/DPS_Quagmaform_Alpha1.safetensors" # Your custom model file URL
+# Download the private model file
+repo_id = "DreamingOracle/Quagmaform_alpha-1"  # Your private repo
+filename = "DPS_Quagmaform_Alpha1.safetensors"  # Your model file
+model_path = hf_hub_download(repo_id=repo_id, filename=filename)
 if torch.cuda.is_available(): 
     torch_dtype = torch.float16
 else: