Update app.py

app.py

@@ -29,7 +29,7 @@ def predict(image):
     with torch.no_grad():
         pred = model(img)
     mask = pred.squeeze(0).squeeze(0).cpu().numpy()
-    mask = (mask > 0.5).astype(np.uint8) * 255  # binarize to 0/255
+    mask = (mask * 255).astype(np.uint8)   # keep gray levels
     return Image.fromarray(mask)
 
 # Gradio interface