Autoencoder / app.py
Aaronsep's picture
Update app.py
5d35a4d verified
raw
history blame
721 Bytes
import gradio as gr
from preprocess import preprocess_single_image # Importamos la función de preprocesamiento
def process_image(img):
"""
Recibe una imagen del usuario y la devuelve en dos versiones:
- Imagen normalizada
- Imagen con ruido agregado
"""
img_clean, img_noisy = preprocess_single_image(img)
return img_clean, img_noisy # Devuelve ambas imágenes
# --- Interfaz de Gradio ---
gr.Interface(fn=process_image,
inputs=gr.Image(type="pil"),
outputs=[gr.Image(type="numpy", label="Imagen Normalizada"),
gr.Image(type="numpy", label="Imagen con Ruido")],
examples=["example1.jpg", "example2.jpg"] # Opcional
).launch()