types

app.py

@@ -45,14 +45,14 @@ def get_seg_mask(img):
 
 def app(img, prompt):
     start_time = datetime.now().timestamp()
-    old_size = img.size
+    old_size = Image.fromarray(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)
-    draw.text((10, 10), f"Old size: {old_size}\nDuration: {int(1000 * (end_time - start_time))}ms\n<{prompt}>", fill=(255, 255, 255))
+    draw.text((50, 10), f"Old size: {old_size}\nDuration: {int(1000 * (end_time - start_time))}ms\n<{prompt}>", fill=(255, 255, 255))
     return overlay_img
 
 #ideally: