Update app.py

app.py

@@ -17,10 +17,10 @@ import os
 import sys
 
 # --- ETAPA 1: Clonar o Repositório do GitHub ---
-repo_name = "SeedVR"
+repo_name = "SeedVR2-3B"
 if not os.path.exists(repo_name):
     print(f"Clonando o repositório {repo_name} do GitHub...")
-    subprocess.run(f"git clone https://github.com/ByteDance-Seed/{repo_name}.git", shell=True, check=True)
+    subprocess.run(f"git clone https://huggingface.co/spaces/ByteDance-Seed/SeedVR2-3B", shell=True, check=True)
 
 # --- ETAPA 2: Mudar para o Diretório e Configurar o Ambiente ---
 os.chdir(repo_name)
@@ -32,8 +32,25 @@ print(f"Diretório atual adicionado ao sys.path.")
 # --- ETAPA 3: Instalar Dependências Corretamente ---
 python_executable = sys.executable
 
-# CORREÇÃO: Instalar o Apex a partir do código-fonte usando --no-build-isolation
-# para que ele possa encontrar o PyTorch já instalado a partir do requirements.txt.
+# CORREÇÃO: Instalar flash-attn e Apex a partir do código-fonte para garantir a compatibilidade.
+
+# Instalar flash-attn
+print("Instalando flash-attn...")
+subprocess.run(
+    [
+        python_executable,
+        "-m",
+        "pip",
+        "install",
+        "flash-attn",
+        "--no-build-isolation"
+    ],
+    check=True
+)
+print("✅ Configuração do flash-attn concluída.")
+
+
+# Instalar Apex
 print("Instalando Apex a partir do código-fonte...")
 subprocess.run(
     [
@@ -41,7 +58,7 @@ subprocess.run(
         "-m",
         "pip",
         "install",
-        "--no-build-isolation",  # Garante que o torch seja encontrado durante a compilação
+        "--no-build-isolation",
         "git+https://github.com/NVIDIA/apex.git"
     ],
     check=True
@@ -207,4 +224,4 @@ with gr.Blocks(title="SeedVR") as demo:
     run_button.click(fn=generation_loop, inputs=[input_file, seed, fps], outputs=[output_image, output_video, download_link])
     
 
-demo.queue().launch(share=True)
+demo.queue().launch(share=True)