Restore chunked PyTorch models at startup

app.py

@@ -11,6 +11,21 @@ from egg_damage.config import load_config
 from egg_damage.gradio_app import build_app
 
 
+def restore_chunked_model(filename: str) -> None:
+    target = ROOT / "models" / filename
+    parts_dir = ROOT / "models" / f"{filename}.parts"
+    if target.exists() or not parts_dir.exists():
+        return
+    part_paths = sorted(parts_dir.glob("*.part"))
+    if not part_paths:
+        return
+    with target.open("wb") as output:
+        for part_path in part_paths:
+            output.write(part_path.read_bytes())
+
+
+restore_chunked_model("mobilenet_v3.pt")
+restore_chunked_model("xception.pt")
 config = load_config(ROOT / "configs" / "space.yaml")
 demo = build_app(config)