feat: patch app.py

app.py

@@ -12,8 +12,6 @@ from addict import Dict
 # 🌟 PARCHE EXTENDIDO PARA EVITAR CRASHEOS EN CPU
 if not torch.cuda.is_available():
     print("Aplicando parches de CPU para ClimateGAN...")
-    
-    # 1. Parche para el temporizador (Timer)
     class DummyEvent:
         def __init__(self, *args, **kwargs): pass
         def record(self, *args, **kwargs): pass
@@ -22,10 +20,8 @@ if not torch.cuda.is_available():
     
     torch.cuda.Event = DummyEvent
     torch.cuda.synchronize = lambda *args, **kwargs: None
-
-    # 2. NUEVO: Parche para interceptar .cuda() forzados en el código fuente
-    # Evita que el SPADEResnetBlock y otros módulos crasheen al instanciarse
     torch.nn.Module.cuda = lambda self, *args, **kwargs: self
+
 # ---------------------------------------------------------------------------
 # CARGA DEL MODELO
 # ---------------------------------------------------------------------------
@@ -35,11 +31,9 @@ try:
     print("Descargando pesos y configuración...")
     hf_token = os.environ.get("HF_TOKEN")
 
-    # 1. Descargamos los pesos
     m_path = hf_hub_download(repo_id="ivanmorals/climategan_weights", filename="masker.pth", token=hf_token)
     p_path = hf_hub_download(repo_id="ivanmorals/climategan_weights", filename="painter.pth", token=hf_token)
     
-    # 2. Descargamos LOS DOS archivos de configuración (YAML)
     opts_m_path = hf_hub_download(repo_id="ivanmorals/climategan_weights", filename="opts_masker.yaml", token=hf_token)
     opts_p_path = hf_hub_download(repo_id="ivanmorals/climategan_weights", filename="opts_painter.yaml", token=hf_token)
 
@@ -50,10 +44,10 @@ try:
     with open(opts_p_path, "r") as f_p:
         opts_painter = Dict(yaml.safe_load(f_p))
 
-    # 3. 🌟 LA FUSIÓN: Injertamos la arquitectura del Painter en la base del Masker
+    # 🌟 LA FUSIÓN: Injertamos la arquitectura del Painter en la base del Masker
     opts.gen.p = opts_painter.gen.p
 
-    # 4. Ajustes forzados para inferencia en Hugging Face
+    # Ajustes forzados para inferencia en Hugging Face
     opts.tasks = ["m", "p", "d", "s"]
     opts.load_paths.m   = m_path
     opts.load_paths.p   = p_path
@@ -68,12 +62,9 @@ try:
     opts.train.kitti.pretrain = False
     opts.train.pseudo.tasks   = []
 
-    # Parche de seguridad: Aseguramos resoluciones y eventos por si los YAML no los traen
-    # Parche de seguridad: Aseguramos resoluciones y eventos por si los YAML no los traen
-    transform_dict = Dict()
-    transform_dict.name = "resize"
-    transform_dict.new_size.default = (640, 640)
-    opts.data.transforms = [transform_dict]
+    # 🌟 SE ELIMINÓ EL PARCHE DE RESOLUCIÓN (opts.data.transforms). EL YAML TIENE EL CONTROL.
+
+    # Parche de seguridad para eventos (Por si acaso el YAML no los trae completos)
     opts.events.smog.airlight = 1.0
     opts.events.smog.beta = 0.1
     opts.events.smog.vr = 10.0