From 51e2ebfc3d598aa5d12b52fbc3d3cbb45d62d1da Mon Sep 17 00:00:00 2001
From: XXXXRT666 <runtongzhong@gmail.com>
Date: Fri, 28 Jun 2024 15:12:29 +0800
Subject: [PATCH] DS_STORE

---
 tools/cmd-denoise.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tools/cmd-denoise.py b/tools/cmd-denoise.py
index 69b51e66..1fdcab6d 100644
--- a/tools/cmd-denoise.py
+++ b/tools/cmd-denoise.py
@@ -1,4 +1,5 @@
 import os,argparse
+import traceback
 
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
@@ -12,7 +13,10 @@ def execute_denoise(input_folder,output_folder):
     # print(input_folder)
     # print(list(os.listdir(input_folder).sort()))
     for name in tqdm(os.listdir(input_folder)):
-        ans("%s/%s"%(input_folder,name),output_path='%s/%s'%(output_folder,name))
+        try:
+            ans("%s/%s"%(input_folder,name),output_path='%s/%s'%(output_folder,name))
+        except:
+            traceback.print_exc()
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()