Update api/seedvr_server.py

api/seedvr_server.py

@@ -28,23 +28,26 @@ except ImportError as e:
     print(f"Verifique se o repositório em '{SEEDVR_REPO_PATH}' está correto e completo.")
     raise e
 
-SEEDVR_ROOT = SEEDVR_REPO_PATH
-CKPTS_ROOT = Path("/data/seedvr_models_fp16") 
-OUTPUT_ROOT = Path(os.getenv("OUTPUT_ROOT", "/app/outputs"))
-INPUT_ROOT = Path(os.getenv("INPUT_ROOT", "/app/inputs"))
-HF_HOME_CACHE = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
-REPO_URL = os.getenv("SEEDVR_GIT_URL", "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler")
-NUM_GPUS_TOTAL = 4#int(os.getenv("NUM_GPUS", "8"))
 
-print("🚀 SeedVRServer (Modo de Chamada Direta) inicializando...")
-for p in [CKPTS_ROOT, OUTPUT_ROOT, INPUT_ROOT, HF_HOME_CACHE]:
-    p.mkdir(parents=True, exist_ok=True)
+class SeedVRServer:
+    
+    SEEDVR_ROOT = SEEDVR_REPO_PATH
+    CKPTS_ROOT = Path("/data/seedvr_models_fp16") 
+    OUTPUT_ROOT = Path(os.getenv("OUTPUT_ROOT", "/app/outputs"))
+    INPUT_ROOT = Path(os.getenv("INPUT_ROOT", "/app/inputs"))
+    HF_HOME_CACHE = Path(os.getenv("HF_HOME", "/data/.cache/huggingface"))
+    REPO_URL = os.getenv("SEEDVR_GIT_URL", "https://github.com/numz/ComfyUI-SeedVR2_VideoUpscaler")
+    NUM_GPUS_TOTAL = 4#int(os.getenv("NUM_GPUS", "8"))
+    
+    print("🚀 SeedVRServer (Modo de Chamada Direta) inicializando...")
+    for p in [CKPTS_ROOT, OUTPUT_ROOT, INPUT_ROOT, HF_HOME_CACHE]:
+        p.mkdir(parents=True, exist_ok=True)
         
-setup_dependencies()
-print("✅ SeedVRServer (Modo de Chamada Direta) pronto.")
+    setup_dependencies()
+    print("✅ SeedVRServer (Modo de Chamada Direta) pronto.")
+
 
 
-class SeedVRServer:
     def __init__(self, **kwargs):
         """
         Inicializa o servidor, define os caminhos e prepara o ambiente.