fix version

app.py

@@ -7,63 +7,19 @@ import subprocess
 import gradio as gr
 from omegaconf import OmegaConf
 
-# ================== 新增的下载逻辑 ==================
-MODEL_REPO = "pandaphd/generative_photography"
-BRANCH = "main"
-LOCAL_DIR = "ckpts"
-
-
-def download_hf_folder():
-    os.makedirs(LOCAL_DIR, exist_ok=True)
-
-    def get_file_list(path=""):
-        api_url = f"https://huggingface.co/{MODEL_REPO}/tree/{BRANCH}/{path}"
-        try:
-            response = requests.get(api_url, timeout=10)
-            response.raise_for_status()
-            return response.json()
-        except Exception as e:
-            raise RuntimeError(f"Failed to get file list: {str(e)}")
-
-    def download_file(remote_path):
-        url = f"https://huggingface.co/{MODEL_REPO}/tree/{BRANCH}/{remote_path}"
-        local_path = os.path.join(LOCAL_DIR, remote_path)
-        os.makedirs(os.path.dirname(local_path), exist_ok=True)
-
-        # 使用 wget 下载（支持断点续传）
-        cmd = [
-            "wget", "-c", "-q", "--show-progress",
-            "-O", local_path, url
-        ]
-        try:
-            subprocess.run(cmd, check=True)
-            print(f"Downloaded: {remote_path}")
-        except subprocess.CalledProcessError:
-            print(f"Failed to download: {remote_path}")
-            raise
-
-    print("Downloading models from Hugging Face...")
-
-    # 递归下载函数
-    def download_recursive(path=""):
-        for item in get_file_list(path):
-            item_path = os.path.join(path, item["path"])
-            if item["type"] == "file":
-                download_file(item_path)
-            elif item["type"] == "directory":
-                download_recursive(item_path)
 
-    try:
-        download_recursive()
-        print("All files downloaded successfully!")
-    except Exception as e:
-        print(f"Critical error during download: {str(e)}")
-        exit(1)
+model_path = "ckpts"
+os.makedirs(model_path, exist_ok=True)
 
+# Download files from Hugging Face repository using wget
+def download_huggingface_files():
+    # URL to the Hugging Face repository
+    repo_url = "https://huggingface.co/pandaphd/generative_photography/tree/main/"
+    # Using wget to download all the files from the given Hugging Face repository
+    subprocess.run(["wget", "-r", "-np", "-nH", "--cut-dirs=3", "-P", model_path, repo_url])
 
-# ================== 执行下载 ==================
-if not os.path.exists(os.path.join(LOCAL_DIR, "models")):
-    download_hf_folder()
+print("Downloading models from Hugging Face...")
+download_huggingface_files()
 
 
 torch.manual_seed(42)