Update app.py

app.py

@@ -31,13 +31,19 @@ def load_models():
     try:
         print("Loading CyberRealistic Pony models...")
         
-        # Use Hugging Face Hub download with minimal local storage
-        print(f"Accessing model from {MODEL_REPO}...")
+        # Download model file using huggingface_hub
+        print(f"Downloading model from {MODEL_REPO}...")
+        model_path = hf_hub_download(
+            repo_id=MODEL_REPO,
+            filename=MODEL_FILENAME,
+            cache_dir="/tmp/hf_cache"  # Use tmp for Spaces
+        )
+        print(f"Model downloaded to: {model_path}")
         
         # Load Text2Img pipeline
         if txt2img_pipe is None:
             txt2img_pipe = StableDiffusionXLPipeline.from_single_file(
-                f"https://huggingface.co/{MODEL_REPO}/resolve/main/{MODEL_FILENAME}",
+                model_path,
                 torch_dtype=torch.float16 if device == "cuda" else torch.float32,
                 use_safetensors=True,
                 variant="fp16" if device == "cuda" else None