Update app.py

app.py

@@ -11,21 +11,10 @@
 # // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # // See the License for the specific language governing permissions and
 # // limitations under the License.
-import torch
-import torch.distributed as dist
+import spaces
+import subprocess
 import os
-import gc
-import logging
 import sys
-import subprocess
-from pathlib import Path
-from urllib.parse import urlparse
-from torch.hub import download_url_to_file
-import gradio as gr
-import mediapy
-from einops import rearrange
-import shutil
-from omegaconf import OmegaConf
 
 # --- ETAPA 1: Preparação do Ambiente ---
 # Clonar o repositório para garantir que todas as pastas de código (data, common, etc.) existam.
@@ -52,6 +41,7 @@ print("Instalando dependências a partir do requirements.txt (isso inclui o PyTo
 subprocess.run([python_executable, "-m", "pip", "install", "-r", "requirements.txt"], check=True)
 print("✅ Dependências básicas (incluindo PyTorch) instaladas.")
 
+
 # **Compilar dependências otimizadas para a GPU L40S (Ada Lovelace)**
 print("Instalando flash-attn compilando do zero...")
 subprocess.run([python_executable, "-m", "pip", "install", "--force-reinstall", "--no-cache-dir", "flash-attn"], check=True)
@@ -61,12 +51,14 @@ if not os.path.exists("apex"):
     subprocess.run("git clone https://github.com/NVIDIA/apex", shell=True, check=True)
 
 # Instala o Apex a partir da fonte clonada. Agora o PyTorch já existe e a compilação funcionará.
+# As flags --cpp_ext e --cuda_ext são essenciais para a compilação.
 subprocess.run(
     [python_executable, "-m", "pip", "install", "-v", "--disable-pip-version-check", "--no-cache-dir", "--global-option=--cpp_ext", "--global-option=--cuda_ext", "./apex"],
     check=True
 )
 print("✅ Configuração do Apex concluída.")
 
+
 # **Download dos modelos e dados de exemplo**
 import torch
 from pathlib import Path
@@ -107,6 +99,7 @@ torch.hub.download_url_to_file('https://huggingface.co/datasets/Iceclear/SeedVR_
 torch.hub.download_url_to_file('https://huggingface.co/datasets/Iceclear/SeedVR_VideoDemos/resolve/main/seedvr_videos_crf23/aigc1k/2_1_lq.mp4', '03.mp4')
 print("✅ Modelos e dados de exemplo baixados.")
 
+
 # --- ETAPA 4: Execução do Código Principal da Aplicação ---
 import mediapy
 from einops import rearrange