.

scripts/process_utils.py

@@ -45,8 +45,8 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
 
     print(f"\nDevice: {device}, Local model: {_use_local}\n")
 
-    # init_model(use_local)
-    # model = load_wd14_tagger_model()
+    init_model(use_local)
+    model = load_wd14_tagger_model()
     # sotai_gen_pipe = initialize_sotai_model()
     # refine_gen_pipe = initialize_refine_model()