From 1b4c01e860168d25a286787a58f8720ea9039cb7 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Fri, 28 Jun 2024 22:39:29 +0800
Subject: [PATCH] DS_STORE (#1258)

---
 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 69b51e6..1fdcab6 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()