Show all prototype DB .pt files in dropdown

webui_gradio.py

@@ -186,10 +186,22 @@ def _list_ckpt_files(folder: Path) -> List[str]:
 
 
 def _list_proto_files(folder: Path) -> List[str]:
+    """
+    List prototype DB candidates.
+
+    On Spaces, users may upload prototype DBs with arbitrary names. We therefore:
+    - include all *.pt in checkpoints_style
+    - but try to exclude obvious training checkpoints like stageX_epochY.pt
+    """
     files = _list_pt_files(folder)
-    # heuristics: prototype db files usually contain "proto" in filename
-    protos = [f for f in files if "proto" in Path(f).name.lower()]
-    return protos if protos else files
+    out: List[str] = []
+    for f in files:
+        name = Path(f).name.lower()
+        # exclude training checkpoints
+        if ("stage" in name) and ("epoch" in name):
+            continue
+        out.append(f)
+    return out if out else files
 
 
 def _guess_default_ckpt(files: List[str]) -> Optional[str]: