types

app.py

@@ -48,6 +48,7 @@ def app(img, prompt):
     old_size = img.size
     img = np.array(Image.fromarray(img).resize(working_size))
     mask = ban_cars_mask[get_seg_mask(img)]
+    mask = mask * 255
     overlay_img = Image.fromarray(np.stack([img[:, :, 0], mask, img[:,:,2]], axis=-1))
     end_time = datetime.now().timestamp()
     draw = ImageDraw.Draw(overlay_img)