New logic

app.py

@@ -55,18 +55,17 @@ def full_pipeline(image, prompt, scale_factor=2.0):
         source = np.array(image) / 255.0
         target_shape = (int(image.height * scale_factor), int(image.width * scale_factor))
 
-        # Preparar parâmetros para JAX
         source_jax = jax.device_put(source, JAX_DEVICE)
         t = jnp.array([1.0 / (scale_factor ** 2)], dtype=jnp.float32)
 
-        # Processar com Thera
+        # Chamada corrigida sem 'do_ensemble'
         upscaled = model_edsr.apply(
-            params_edsr,
+            {'params': params_edsr},  # Estrutura de parâmetros correta
             source_jax,
             t,
-            target_shape,
-            do_ensemble=True
+            target_shape
         )
+
         upscaled_pil = Image.fromarray((np.array(upscaled) * 255).astype(np.uint8))
 
         # 2. Gerar Bas-Relief