Update app.py
Browse files
app.py
CHANGED
@@ -49,4 +49,4 @@ transform = transforms.Compose([
|
|
49 |
def detectar_anomalia(imagen):
|
50 |
with torch.no_grad():
|
51 |
img_tensor = transform(imagen).unsqueeze(0) # Añadir batch
|
52 |
-
reconstruida = model(img_tensor).squeeze(0).squeeze(
|
|
|
49 |
def detectar_anomalia(imagen):
|
50 |
with torch.no_grad():
|
51 |
img_tensor = transform(imagen).unsqueeze(0) # Añadir batch
|
52 |
+
reconstruida = model(img_tensor).squeeze(0).squeeze(0)
|