Bug fix: a error is raised when refresh file display if there is a fo…

…ld in the work_dir
zh3305 · Nov 3, 2023 · c869259 · c869259
1 parent 62370dd
commit c869259
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/web_ui.py b/src/web_ui.py
@@ -67,9 +67,9 @@ def refresh_file_display(state_dict: Dict) -> List[str]:
     filenames = os.listdir(work_dir)
     paths = []
     for filename in filenames:
-        paths.append(
-            os.path.join(work_dir, filename)
-        )
+        path = os.path.join(work_dir, filename)
+        if not os.path.isdir(path):
+            paths.append(path)
     return paths