Update app.py

app.py

@@ -1,7 +1,14 @@
 import gradio as gr
+import tensorflow as tf
+import numpy as np
+import cv2  # Faltaba importar OpenCV
 from huggingface_hub import hf_hub_download
+
+# Descargar modelo desde Hugging Face
 model_path = hf_hub_download(repo_id="Bmo411/DenoisingAutoencoder", filename="autoencoder_complete_model_Fourier.keras")
-import tensorflow as tf
+
+# Cargar el modelo una sola vez para evitar recargas en cada ejecución
+model = tf.keras.models.load_model(model_path, compile=True)
 
 def degrade_image(image, downscale_factor=4):
     """Reduce la calidad de la imagen reduciendo su tamaño y volviéndola a escalar."""
@@ -35,10 +42,31 @@ def preprocess_image(image, std_dev=0.1, downscale_factor=4, target_size=(256, 2
     return noisy_img
 
 def Denoiser(imagen):
-    model = tf.keras.models.load_model("autoencoder_complete_model_Fourier.keras", compile=True) 
-    image = preprocess_image(imagen)
-    reconstruct = model.predict(image)
-    return image,recostruct 
+    """Aplica el modelo autoencoder para eliminar ruido de la imagen."""
     
-demo = gr.Interface(fn=Denoiser, inputs="image", outputs=["image","recostruct"])
-demo.launch()   
+    # Convertir imagen de entrada a array NumPy y asegurarse de que está en formato correcto
+    imagen = np.array(imagen)  # Convertir de PIL a NumPy
+
+    # Preprocesar imagen (degradarla y agregar ruido)
+    noisy_image = preprocess_image(imagen)
+
+    # Expandir dimensiones para que tenga el formato correcto para el modelo (batch_size, h, w, c)
+    noisy_image_input = np.expand_dims(noisy_image, axis=0)
+
+    # Predecir con el autoencoder
+    reconstructed = model.predict(noisy_image_input)[0]  # Quitar batch_size
+
+    return noisy_image, reconstructed
+
+# Crear interfaz en Gradio con dos salidas de imagen
+demo = gr.Interface(
+    fn=Denoiser, 
+    inputs=gr.Image(type="numpy"),  # Para que Gradio pase la imagen como un array de NumPy
+    outputs=[gr.Image(type="numpy", label="Imagen con Ruido"), 
+             gr.Image(type="numpy", label="Imagen Restaurada")],
+    title="Autoencoder para Denoising",
+    description="Este modelo de autoencoder reduce el ruido en imágenes. Sube una imagen y el modelo generará una versión restaurada."
+)
+
+# Lanzar la aplicación en Gradio
+demo.launch()