Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (VERSÃO FINAL E CORRIGIDA)
+# app.py (VERSÃO FINAL COM LAZY IMPORT)
 
 import gradio as gr
 import os
@@ -7,19 +7,25 @@ import shutil
 import subprocess
 import mimetypes
 from pathlib import Path
-from torch.hub import download_url_to_file
+# A importação do torch.hub foi REMOVIDA daqui
 
 # --- BLOCO DE CONFIGURAÇÃO E DOWNLOAD DE MODELO ---
 APP_DIR = "/app"; SEEDVR_DIR = os.path.join(APP_DIR, "SeedVR")
 MODEL_CACHE_DIR = "/tmp/models"; CKPTS_DIR = os.path.join(MODEL_CACHE_DIR, "ckpts")
 os.makedirs(CKPTS_DIR, exist_ok=True)
+
+print("Verificando e baixando modelos para /tmp/models/ckpts...")
+# !!! A MUDANÇA CRUCIAL ESTÁ AQUI !!!
+# A importação é feita apenas quando necessária.
+from torch.hub import download_url_to_file
+
 files_to_download = {
     "seedvr2_ema_3b.pth": "https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/seedvr2_ema_3b.pth",
     "ema_vae.pth": "https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/ema_vae.pth",
     "pos_emb.pt": "https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/pos_emb.pt",
     "neg_emb.pt": "https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/neg_emb.pt",
 }
-print("Verificando e baixando modelos para /tmp/models/ckpts...")
+
 for filename, url in files_to_download.items():
     destination_path = os.path.join(CKPTS_DIR, filename)
     if not os.path.exists(destination_path):
@@ -28,6 +34,7 @@ for filename, url in files_to_download.items():
 print("Verificação de modelos concluída.")
 # --------------------------------------------------------------------
 
+# O resto do código (run_inference, UI do Gradio) permanece o mesmo.
 def run_inference(video_path, seed, res_h, res_w):
     if video_path is None: raise gr.Error("Por favor, faça o upload de um arquivo.")
     job_id = str(uuid.uuid4()); input_dir = os.path.join("/tmp", "temp_inputs", job_id); output_dir = os.path.join("/tmp", "temp_outputs", job_id)
@@ -65,7 +72,6 @@ def run_inference(video_path, seed, res_h, res_w):
         shutil.rmtree(input_dir, ignore_errors=True)
         if os.path.exists(patched_script_path): os.remove(patched_script_path)
 
-# --- Interface Gráfica Gradio ---
 with gr.Blocks(css="footer {display: none !important}") as demo:
     gr.Markdown("# 🚀 Interface de Inferência para SeedVR2")
     gr.Markdown("Faça o upload de um vídeo ou imagem, ajuste os parâmetros e clique em 'Executar'.")
@@ -83,9 +89,7 @@ with gr.Blocks(css="footer {display: none !important}") as demo:
             output_image = gr.Image(label="Saída de Imagem")
             output_video = gr.Video(label="Saída de Vídeo")
             log_box = gr.Textbox(label="Logs em Tempo Real", lines=15, autoscroll=True, interactive=False)
-    
-    # !!! A CORREÇÃO FINAL ESTÁ AQUI !!!
-    # Estas duas chamadas foram movidas para DENTRO do bloco 'with gr.Blocks()'.
+            
     run_button.click(
         fn=run_inference,
         inputs=[input_media, seed, res_h, res_w],