New logic

app.py

@@ -25,7 +25,9 @@ def load_thera_model(repo_id, filename):
     model_path = hf_hub_download(repo_id=repo_id, filename=filename)
     with open(model_path, 'rb') as fh:
         check = pickle.load(fh)
-        params, backbone, size = check['model'], check['backbone'], check['size']
+        # Ajustar a estrutura dos parâmetros
+        params = check['model']['params']  # Acessar os parâmetros corretamente
+        backbone, size = check['backbone'], check['size']
     model = build_thera(3, backbone, size)
     return model, params
 
@@ -58,9 +60,9 @@ def full_pipeline(image, prompt, scale_factor=2.0):
         source_jax = jax.device_put(source, JAX_DEVICE)
         t = jnp.array([1.0 / (scale_factor ** 2)], dtype=jnp.float32)
 
-        # Chamada corrigida sem 'do_ensemble'
+        # Chamada corrigida com estrutura de parâmetros adequada
         upscaled = model_edsr.apply(
-            {'params': params_edsr},  # Estrutura de parâmetros correta
+            params_edsr,  # Parâmetros já estruturados corretamente
             source_jax,
             t,
             target_shape
@@ -125,4 +127,4 @@ with gr.Blocks(title="Super Res + Bas-Relief") as app:
     )
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch(share=False)  # Ativando link público