Update app.py

app.py

@@ -5,6 +5,38 @@ import os
 import numpy as np
 from PIL import Image
 from omegaconf import OmegaConf
+from huggingface_hub import hf_hub_download
+
+# --- Início: Bloco de Download Automático do Modelo ---
+
+# Define o diretório e o caminho para os pesos do modelo
+WEIGHTS_DIR = "./pretrained_weights/ByteMorpher"
+MODEL_FILENAME = "dit.safetensors"
+MODEL_PATH = os.path.join(WEIGHTS_DIR, MODEL_FILENAME)
+
+# Cria o diretório se ele não existir
+os.makedirs(WEIGHTS_DIR, exist_ok=True)
+
+# Verifica se o modelo já existe antes de fazer o download
+if not os.path.exists(MODEL_PATH):
+    print(f"Modelo não encontrado em {MODEL_PATH}. Baixando do Hugging Face Hub...")
+    try:
+        hf_hub_download(
+            repo_id="ByteDance-Seed/BM-Model",
+            filename=MODEL_FILENAME,
+            local_dir=WEIGHTS_DIR,
+            local_dir_use_symlinks=False  # Recomendado para Hugging Face Spaces
+        )
+        print("Download do modelo concluído com sucesso.")
+    except Exception as e:
+        print(f"Ocorreu um erro durante o download do modelo: {e}")
+        # Se o download falhar, o aplicativo não poderá funcionar.
+        # Você pode adicionar um tratamento de erro mais robusto aqui se desejar.
+else:
+    print(f"Modelo já existe em {MODEL_PATH}. Pulando o download.")
+
+# --- Fim: Bloco de Download Automático do Modelo ---
+
 
 from image_datasets.dataset import image_resize
 args = OmegaConf.load("inference_configs/inference.yaml")
@@ -19,10 +51,11 @@ def generate(image: Image.Image, edit_prompt: str):
     
     global sampler
     if sampler == None:
+        # A inicialização do sampler agora ocorrerá após a confirmação de que o modelo foi baixado.
         sampler = XFluxSampler(
             device = device,
             ip_loaded=False,
-            spatial_condition=True,
+            spatial_condition=False,
             clip_image_processor=None,
             image_encoder=None,
             improj=None,
@@ -202,4 +235,5 @@ def create_app():
     return app
 
 if __name__ == "__main__":
-    create_app().launch(debug=False, share=False, ssr_mode=False)
+    app = create_app()
+    app.launch(debug=False, share=False)