Update app.py

app.py

@@ -39,7 +39,7 @@ def greet(image):
 
 
     with torch.no_grad():
-        mask_pred = model(image)
+        mask_pred = torch.sigmoid(model(image))
         
     return mask_pred[0].numpy()