Spaces:

paascorb
/

practica3

Runtime error

paascorb commited on Apr 9, 2023

Commit

a186b92

1 Parent(s): f348f34

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,9 +26,9 @@ def predict(img):
     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
-    learner.to(device)
     with torch.no_grad():
-        outputs = learner(tensor)
     outputs = torch.argmax(outputs,1)
     mask = np.array(outputs.cpu())

     image = transforms.Resize((480,640))(img)
     tensor = transform_image(image=image)
+    model.to(device)
     with torch.no_grad():
+        outputs = model(tensor)
     outputs = torch.argmax(outputs,1)
     mask = np.array(outputs.cpu())