更新代码结构

util/download_weights.py

@@ -1,4 +1,3 @@
-import os
 import subprocess
 from pathlib import Path
 

util/omniparser.py

@@ -8,7 +8,6 @@ class Omniparser(object):
     def __init__(self, config: Dict):
         self.config = config
         device = 'cuda' if torch.cuda.is_available() else 'cpu'
-
         self.som_model = get_yolo_model(model_path=config['som_model_path'])
         self.caption_model_processor = get_caption_model_processor(model_name=config['caption_model_name'], model_name_or_path=config['caption_model_path'], device=device)
         print('Omniparser initialized!')

util/tool.py

@@ -1,6 +1,4 @@
 import os
-import logging
-import argparse
 import shlex
 import subprocess
 import threading
@@ -31,7 +29,6 @@ def execute_command(command, shell=False):
                 'returncode': result.returncode
             }
         except Exception as e:
-            logger.error("\n" + traceback.format_exc() + "\n")
             return {
                 'status': 'error',
                 'message': str(e)