Update app.py

app.py

@@ -12,6 +12,52 @@
 # // See the License for the specific language governing permissions and
 # // limitations under the License.
 
+
+
+import spaces
+import subprocess
+
+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.")
+    
+
+# --- ETAPA 1: Clonar o Repositório e Mudar para o Diretório ---
+repo_name = "SeedVR"
+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)
+
+# Garante que estamos no diretório certo
+if not os.getcwd().endswith(repo_name):
+    os.chdir(repo_name)
+
+sys.path.insert(0, os.path.abspath('.'))
+
+
+
 import os
 import torch
 import mediapy