Update app.py

app.py

@@ -21,7 +21,7 @@ def inverse_fourier_transform(x):
     real_part, imag_part = tf.split(x, num_or_size_splits=2, axis=-1)
     complex_fourier = tf.complex(real_part, imag_part)
     return tf.abs(tf.signal.ifft2d(complex_fourier))
-    
+
 def degrade_image(image, downscale_factor=4):
     """Reduce la calidad de la imagen reduciendo su tamaño y volviéndola a escalar."""
     h, w = image.shape[:2]
@@ -68,6 +68,10 @@ def Denoiser(imagen):
     # Predecir con el autoencoder
     reconstructed = model.predict(noisy_image_input)[0]  # Quitar batch_size
 
+    # Asegurarse de que la imagen restaurada esté en el rango [0, 255] para la visualización
+    noisy_image = np.uint8(noisy_image * 255)
+    reconstructed = np.uint8(reconstructed * 255)
+
     return noisy_image, reconstructed
 
 # Crear interfaz en Gradio con dos salidas de imagen
@@ -81,4 +85,4 @@ demo = gr.Interface(
 )
 
 # Lanzar la aplicación en Gradio
-demo.launch()
+demo.launch()