Update api/seedvr_server.py
Browse files- api/seedvr_server.py +9 -0
api/seedvr_server.py
CHANGED
|
@@ -5,6 +5,7 @@ import sys
|
|
| 5 |
import shutil
|
| 6 |
import mimetypes
|
| 7 |
import time
|
|
|
|
| 8 |
from pathlib import Path
|
| 9 |
from typing import Optional, Callable
|
| 10 |
from types import SimpleNamespace
|
|
@@ -112,9 +113,15 @@ class SeedVRServer:
|
|
| 112 |
)
|
| 113 |
|
| 114 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
| 115 |
# Chama a função importada do script original, passando o callback de progresso.
|
|
|
|
| 116 |
result_tensor, original_fps, _, _ = run_inference_logic(args, progress_callback=progress)
|
| 117 |
|
|
|
|
| 118 |
if progress:
|
| 119 |
progress(0.95, "Saving the final video...")
|
| 120 |
|
|
@@ -129,5 +136,7 @@ class SeedVRServer:
|
|
| 129 |
|
| 130 |
except Exception as e:
|
| 131 |
print(f"❌ Error during direct inference execution: {e}")
|
|
|
|
|
|
|
| 132 |
# Propaga o erro para a UI do Gradio, que o exibirá de forma amigável.
|
| 133 |
raise
|
|
|
|
| 5 |
import shutil
|
| 6 |
import mimetypes
|
| 7 |
import time
|
| 8 |
+
import subprocess # Necessário para clonar o repositório
|
| 9 |
from pathlib import Path
|
| 10 |
from typing import Optional, Callable
|
| 11 |
from types import SimpleNamespace
|
|
|
|
| 113 |
)
|
| 114 |
|
| 115 |
try:
|
| 116 |
+
# Informa a UI que o processo começou.
|
| 117 |
+
if progress:
|
| 118 |
+
progress(0.01, "Initializing...")
|
| 119 |
+
|
| 120 |
# Chama a função importada do script original, passando o callback de progresso.
|
| 121 |
+
# Este callback será chamado de dentro da lógica de multi-processamento.
|
| 122 |
result_tensor, original_fps, _, _ = run_inference_logic(args, progress_callback=progress)
|
| 123 |
|
| 124 |
+
# Informa a UI que a inferência terminou e o salvamento vai começar.
|
| 125 |
if progress:
|
| 126 |
progress(0.95, "Saving the final video...")
|
| 127 |
|
|
|
|
| 136 |
|
| 137 |
except Exception as e:
|
| 138 |
print(f"❌ Error during direct inference execution: {e}")
|
| 139 |
+
import traceback
|
| 140 |
+
traceback.print_exc()
|
| 141 |
# Propaga o erro para a UI do Gradio, que o exibirá de forma amigável.
|
| 142 |
raise
|