Update app.py

app.py

@@ -13,7 +13,7 @@ import argparse
 import gradio as gr
 import spaces
 from huggingface_hub import hf_hub_url
-import requests
+import subprocess
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -204,12 +204,15 @@ with gr.Blocks() as demo:
    
 
 
-def download_and_save_model(model_name_or_path, save_directory):
-    from transformers import AutoModel
-    model = AutoModel.from_pretrained(model_name_or_path)
-    model.save_pretrained(save_directory)
-    print(f"Model saved to {save_directory}", model_name_or_path)
+def download_with_wget(url, save_path):
+    
+    try:
+        subprocess.run(['wget', url, '-O', save_path], check=True)
+        print(f"Downloaded to {save_path}")
+    except subprocess.CalledProcessError as e:
+        print(f"Error downloading file: {e}")
 def download_model():
+    
     urls = [
         'https://huggingface.co/stabilityai/StableWurst/resolve/main/stage_a.safetensors',
         'https://huggingface.co/stabilityai/StableWurst/resolve/main/previewer.safetensors',
@@ -219,7 +222,7 @@ def download_model():
         'https://huggingface.co/roubaofeipi/UltraPixel/blob/main/ultrapixel_t2i.safetensors'  
     ]
     for file_url in urls:
-        download_and_save_model(file_url, 'models')        
+        download_with_wget(file_url, 'models')        
 if __name__ == "__main__":
    
     args = parse_args()