update

app.py

@@ -96,7 +96,7 @@ if __name__ == "__main__":
     server_port = int(SERVER_PORT)
     root_path = ROOT_PATH
     elo_results_dir = ELO_RESULTS_DIR
-    models = ModelManager(enable_nsfw=True, pre_download=True, debug_packages=True)
+    models = ModelManager(enable_nsfw=False, do_pre_download=True, do_debug_packages=True)
     # models = ModelManager(enable_nsfw=False, pre_download=False, debug_packages=False)
     
     elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)

model/model_manager.py

@@ -19,7 +19,7 @@ def debug_packages():
         print(f"{package.key}=={package.version}")
         
 class ModelManager:
-    def __init__(self, enable_nsfw=False, pre_download=False, debug_packages=False):
+    def __init__(self, enable_nsfw=False, do_pre_download=False, do_debug_packages=False):
         self.model_ig_list = IMAGE_GENERATION_MODELS
         self.model_ie_list = IMAGE_EDITION_MODELS
         self.model_vg_list = VIDEO_GENERATION_MODELS
@@ -28,9 +28,9 @@ class ModelManager:
         self.enable_nsfw = enable_nsfw
         self.load_guard(enable_nsfw)
         self.loaded_models = {}
-        if pre_download:
+        if do_pre_download:
             pre_download_all_models()
-        if debug_packages:
+        if do_debug_packages:
             debug_packages()
 
     def load_model_pipe(self, model_name):