Update app.py
Browse files
app.py
CHANGED
@@ -57,3 +57,16 @@ def detectar_anomalia(imagen):
|
|
57 |
img_tensor = transform(imagen).unsqueeze(0) # A帽adir batch
|
58 |
reconstruida = model(img_tensor).squeeze(0).squeeze(0)
|
59 |
return reconstruida.numpy() # Convertir a numpy para visualizaci贸n
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
img_tensor = transform(imagen).unsqueeze(0) # A帽adir batch
|
58 |
reconstruida = model(img_tensor).squeeze(0).squeeze(0)
|
59 |
return reconstruida.numpy() # Convertir a numpy para visualizaci贸n
|
60 |
+
|
61 |
+
|
62 |
+
# Interfaz de Gradio
|
63 |
+
interface = gr.Interface(
|
64 |
+
fn=detectar_anomalia,
|
65 |
+
inputs=gr.Image(type="pil"),
|
66 |
+
outputs=[gr.Image(type="numpy"), gr.Text()],
|
67 |
+
title="Detecci贸n de Anomal铆as con Autoencoder",
|
68 |
+
description="Sube una imagen para detectar anomal铆as usando un autoencoder entrenado."
|
69 |
+
)
|
70 |
+
|
71 |
+
interface.launch()
|
72 |
+
|