Add & use modules.errors.print_error where currently printing exception info by hand

1 files changed, 2 insertions, 4 deletions

diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py
index 0131dea4..d2f647fe 100644
--- a/modules/gfpgan_model.py
+++ b/modules/gfpgan_model.py
@@ -1,12 +1,11 @@
 import os

-import sys

-import traceback

 

 import facexlib

 import gfpgan

 

 import modules.face_restoration

 from modules import paths, shared, devices, modelloader

+from modules.errors import print_error

 

 model_dir = "GFPGAN"

 user_path = None

@@ -112,5 +111,4 @@ def setup_model(dirname):
 

         shared.face_restorers.append(FaceRestorerGFPGAN())

     except Exception:

-        print("Error setting up GFPGAN:", file=sys.stderr)

-        print(traceback.format_exc(), file=sys.stderr)

+        print_error("Error setting up GFPGAN", exc_info=True)