EuuIia commited on
Commit
799e694
·
verified ·
1 Parent(s): a303286

Update api/seedvr_server.py

Browse files
Files changed (1) hide show
  1. 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