Update app.py

app.py

@@ -14,6 +14,7 @@ import gradio as gr
 import spaces
 from huggingface_hub import hf_hub_url
 import subprocess
+from huggingface_hub import hf_hub_download
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -219,10 +220,13 @@ def download_model():
         'https://huggingface.co/stabilityai/StableWurst/resolve/main/effnet_encoder.safetensors',
         'https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_b_lite_bf16.safetensors', 
         'https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_c_bf16.safetensors', 
-        'https://huggingface.co/roubaofeipi/UltraPixel/blob/main/ultrapixel_t2i.safetensors'  
+        
     ]
     for file_url in urls:
-        download_with_wget(file_url, os.path.join('models', file_url.split('/')[-1]))        
+        hf_hub_download(repo_id="stabilityai/stable-cascade", filename=file_url.split('/')[-1], local_dir='models')
+    #   'https://huggingface.co/roubaofeipi/UltraPixel/blob/main/ultrapixel_t2i.safetensors'  
+    hf_hub_download(repo_id="roubaofeipi/UltraPixel", filename='ultrapixel_t2i.safetensors', local_dir='models')
+    
 if __name__ == "__main__":
    
     args = parse_args()