Update app.py

app.py

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