Fix #31 app.py

app.py

@@ -19,7 +19,7 @@ def resize(img, size):
     return F.interpolate(img, size=size, mode='bilinear', align_corners=False)
 
 def denormalize(x):
-    return (x + 1) / 2
+    return (x + 1) / 2  # Valores en [0, 1]
 
 # Definición de las clases de los modelos (Generator, StyleEncoder, MappingNetwork, ResBlk, AdaIN, AdainResBlk)
 class ResBlk(nn.Module):
@@ -280,8 +280,8 @@ class Solver(object):
             generated_image = self.G(source_image, s_ref)
 
             # Denormalizar la imagen para mostrarla en la interfaz
-            generated_image = denormalize(generated_image.squeeze(0)).cpu().numpy()
-            return generated_image
+            generated_image = denormalize(generated_image.squeeze(0)).cpu()
+            return (generated_image * 255).clamp(0, 255).byte().permute(1, 2, 0).numpy() # Convertir a NumPy y a rango válido
 
 # Función principal para la inferencia
 def main(source_image, reference_image, checkpoint_path, args):