types

app.py

@@ -46,7 +46,7 @@ def get_seg_mask(img):
 def app(img, prompt):
     start_time = datetime.now().timestamp()
     old_size = img.size
-    img = Image.fromarray(img).resize(working_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))
     end_time = datetime.now().timestamp()