Update app.py

app.py

@@ -10,6 +10,7 @@ import matplotlib.pyplot as plt
 from roboflow import Roboflow
 from PIL import Image
 import torch
+from string import Template  # <-- NEW
 
 # Quiet some noisy libs on Spaces (harmless locally)
 os.environ.setdefault("YOLO_CONFIG_DIR", "/tmp/Ultralytics")
@@ -144,7 +145,7 @@ def parse_roboflow_url(s: str):
             version = None
             if len(p) >= 3:
                 v = p[2]
-                if v.lower().startsWith('v') and v[1:].isdigit():
+                if v.lower().startswith('v') and v[1:].isdigit():  # <-- FIXED
                     version = int(v[1:])
                 elif v.isdigit():
                     version = int(v)
@@ -470,53 +471,54 @@ def _install_workspace_shim_v3(dest_dir: str, module_default: str = "rtdetrv2_py
     os.makedirs(dest_dir, exist_ok=True)
     sc_path = os.path.join(dest_dir, "sitecustomize.py")
 
-    # NOTE: Not an f-string. Escape literal braces with {{ }} and only format {module_default}.
-    code = textwrap.dedent("""\
-        import os, importlib, types
-        try:
-            mod_default = os.environ.get("RTDETR_PYMODULE", "{module_default}") or "{module_default}"
-            ws_mod = importlib.import_module("rtdetrv2_pytorch.src.core.workspace")
-            _orig_create = ws_mod.create
+    # Use Template so braces in code remain literal; only $module_default is substituted.
+    tmpl = Template(r"""
+import os, importlib, types
+try:
+    mod_default = os.environ.get("RTDETR_PYMODULE", "$module_default") or "$module_default"
+    ws_mod = importlib.import_module("rtdetrv2_pytorch.src.core.workspace")
+    _orig_create = ws_mod.create
 
-            def _ensure_pymodule_object(cfg):
-                pm = None
-                try:
-                    pm = cfg.get("_pymodule", None)
-                except Exception:
-                    pm = None
-                if isinstance(pm, str) or pm is None:
-                    name = pm.strip() if isinstance(pm, str) and pm.strip() else mod_default
-                    try:
-                        mod = importlib.import_module(name)
-                    except Exception:
-                        mod = importlib.import_module(mod_default)
-                    try:
-                        cfg["_pymodule"] = mod
-                    except Exception:
-                        pass
-                    return mod
-                if isinstance(pm, types.ModuleType):
-                    return pm
-                try:
-                    mod = importlib.import_module(mod_default)
-                    cfg["_pymodule"] = mod
-                    return mod
-                except Exception:
-                    return pm
+    def _ensure_pymodule_object(cfg):
+        pm = None
+        try:
+            pm = cfg.get("_pymodule", None)
+        except Exception:
+            pm = None
+        if isinstance(pm, str) or pm is None:
+            name = pm.strip() if isinstance(pm, str) and pm.strip() else mod_default
+            try:
+                mod = importlib.import_module(name)
+            except Exception:
+                mod = importlib.import_module(mod_default)
+            try:
+                cfg["_pymodule"] = mod
+            except Exception:
+                pass
+            return mod
+        if isinstance(pm, types.ModuleType):
+            return pm
+        try:
+            mod = importlib.import_module(mod_default)
+            cfg["_pymodule"] = mod
+            return mod
+        except Exception:
+            return pm
 
-            def create(name, **kwargs):
-                cfg = kwargs.get("cfg")
-                if not isinstance(cfg, dict):
-                    cfg = {} if cfg is None else dict(cfg)
-                    kwargs["cfg"] = cfg
-                _ensure_pymodule_object(cfg)
-                return _orig_create(name, **kwargs)
+    def create(name, **kwargs):
+        cfg = kwargs.get("cfg")
+        if not isinstance(cfg, dict):
+            cfg = {} if cfg is None else dict(cfg)
+            kwargs["cfg"] = cfg
+        _ensure_pymodule_object(cfg)
+        return _orig_create(name, **kwargs)
 
-            ws_mod.create = create
-        except Exception:
-            # never block training if shim fails
-            pass
-    """).format(module_default=module_default)
+    ws_mod.create = create
+except Exception:
+    # never block training if shim fails
+    pass
+""")
+    code = tmpl.substitute(module_default=module_default)
 
     with open(sc_path, "w", encoding="utf-8") as f:
         f.write(code)
@@ -694,7 +696,7 @@ def patch_base_config(base_cfg_path, merged_dir, class_count, run_name,
                 "shuffle": bool(default_shuffle),
                 "num_workers": 2,
                 "drop_last": bool(dl_key == "train_dataloader"),
-                "collate_fn": {"type": "BatchImageCollateFuncion"},
+                "collate_fn": {"type": "BatchImageCollateFunction"},  # <-- FIXED name
                 "total_batch_size": int(batch),
             }
             cfg[dl_key] = block