JeanCGuerrero commited on
Commit
f2c9943
verified
1 Parent(s): de04d07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
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
+