Upload demo/publish_space.py with huggingface_hub

demo/publish_space.py

@@ -24,25 +24,21 @@ STATIC_FILE_MAP = {
     PROJECT_ROOT / "infer" / "common_infer.py": "infer/common_infer.py",
     PROJECT_ROOT / "infer" / "infer.py": "infer/infer.py",
     PROJECT_ROOT / "infer" / "infer.yaml": "infer/infer.yaml",
+    PROJECT_ROOT / "models" / "__init__.py": "models/__init__.py",
+    PROJECT_ROOT / "models" / "mmdit.py": "models/mmdit.py",
+    PROJECT_ROOT / "models" / "multiLayer_adapter.py": "models/multiLayer_adapter.py",
+    PROJECT_ROOT / "models" / "pipeline.py": "models/pipeline.py",
+    PROJECT_ROOT / "models" / "transp_vae.py": "models/transp_vae.py",
+    PROJECT_ROOT / "tools" / "__init__.py": "tools/__init__.py",
+    PROJECT_ROOT / "tools" / "tools.py": "tools/tools.py",
 }
 
-DIRECTORIES_TO_UPLOAD = [
-    PROJECT_ROOT / "models",
-    PROJECT_ROOT / "tools",
-]
-
 
 def iter_uploads():
     for local_path, remote_path in STATIC_FILE_MAP.items():
         if local_path.exists():
             yield local_path, remote_path
 
-    for directory in DIRECTORIES_TO_UPLOAD:
-        if not directory.exists():
-            continue
-        for local_path in directory.rglob("*.py"):
-            yield local_path, str(local_path.relative_to(PROJECT_ROOT))
-
 
 def main():
     parser = argparse.ArgumentParser(