Update app.py

app.py

@@ -1,6 +1,33 @@
 import os
 import sys
 import subprocess
+import importlib.util
+
+# --- ETAPA 0: Instalação Final do flash-attn ---
+
+# Verifica se o flash_attn já está instalado. Se não, instala.
+package_name = 'flash_attn'
+spec = importlib.util.find_spec(package_name)
+if spec is None:
+    print(f"Instalando o pacote que faltava: {package_name}. Isso pode levar um minuto...")
+    # Usamos o python executável do ambiente atual para instalar o pacote
+    python_executable = sys.executable
+    subprocess.run(
+        [
+            python_executable, "-m", "pip", "install", 
+            "flash_attn==2.5.9.post1", 
+            "--no-build-isolation"
+        ],
+        check=True
+    )
+    print(f"✅ {package_name} instalado com sucesso.")
+else:
+    print(f"✅ Pacote {package_name} já está instalado.")
+
+
+# A partir daqui, o ambiente está 100% pronto.
+# ---------------------------------------------------------------------
+
 import spaces
 from pathlib import Path
 from urllib.parse import urlparse
@@ -41,7 +68,7 @@ from common.seed import set_seed
 from projects.video_diffusion_sr.infer import VideoDiffusionInfer
 from common.distributed.ops import sync_data
 
-print("Ambiente Conda carregado com sucesso. Iniciando a aplicação...")
+print("Ambiente Conda carregado e verificado. Iniciando a aplicação...")
 
 # --- ETAPA 2: Baixar os Modelos Pré-treinados ---
 print("Baixando modelos pré-treinados...")
@@ -161,4 +188,4 @@ with gr.Blocks(title="SeedVR") as demo:
     download_link = gr.File(label="Baixar Resultado")
     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) I