lsb commited on
Commit
80d1c90
1 Parent(s): 11fb5ff
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -53,12 +53,9 @@ def app(img, prompt):
53
  start_time = datetime.now().timestamp()
54
  old_size = Image.fromarray(img).size
55
  img = np.array(Image.fromarray(img).resize(working_size))
56
- mask = ban_cars_mask[get_seg_mask(img)]
57
- mask = mask * 255
58
  mask_time = datetime.now().timestamp()
59
- print(prompt)
60
- print(img)
61
- print(mask)
62
  overlay_img = inpainting_pipeline(
63
  prompt=prompt,
64
  image=img,
 
53
  start_time = datetime.now().timestamp()
54
  old_size = Image.fromarray(img).size
55
  img = np.array(Image.fromarray(img).resize(working_size))
56
+ mask = ban_cars_mask[get_seg_mask(img)] * 255
 
57
  mask_time = datetime.now().timestamp()
58
+ print(prompt.__class__, img.__class__, mask.__class__, img.shape, mask.shape)
 
 
59
  overlay_img = inpainting_pipeline(
60
  prompt=prompt,
61
  image=img,