Upload app.py

app.py

@@ -40,7 +40,11 @@ async def segment(req: SegmentRequest):
         inputs["original_sizes"].cpu(),
         inputs["reshaped_input_sizes"].cpu()
     )
-    mask_np = (masks[0][0] * 255).astype("uint8")
+    mask_tensor = masks[0][0]                # torch.FloatTensor
+    mask_np     = (mask_tensor * 255)        \
+                    .to(torch.uint8)       \
+                    .cpu()                 \
+                    .numpy()               # now uint8 numpy array
     buf = io.BytesIO()
     Image.fromarray(mask_np).save(buf, format="PNG")
     buf.seek(0)