diff --git a/src/web_ui.py b/src/web_ui.py index e5df941..c406a24 100644 --- 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