types? f8

app.py

@@ -48,7 +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)]
-    overlay_img = Image.fromarray(np.stack([img[:, :, 0], mask / 2, img[:,:,2]], axis=-1))
+    overlay_img = Image.fromarray(np.stack([img[:, :, 0], mask, img[:,:,2]], axis=-1))
     end_time = datetime.now().timestamp()
     draw = ImageDraw.Draw(overlay_img)
     draw.text((10, 10), f"Old size: {old_size}\nDuration: {int(1000 * (end_time - start_time))}ms\n<{prompt}>", fill=(255, 255, 255))