max commited on
Commit
0f09c7c
1 Parent(s): 3746f14

removed debug code

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -301,7 +301,7 @@ class Predictor:
301
  )
302
  i.putalpha(morig)
303
  img = i
304
- img.save('0.png')
305
  assert img.width == img.height
306
  assert img.width > 512 and img.width < 512*2
307
 
@@ -319,7 +319,7 @@ class Predictor:
319
 
320
  for ix, tc in enumerate(tile_coords(img, n=2)):
321
  i = img.crop(tc)
322
- i.save(f't{ix}.png')
323
  m = i.getchannel('A')
324
 
325
  """Run a single prediction on the model"""
@@ -338,7 +338,7 @@ class Predictor:
338
  inpainted,
339
  1-(np.array(m) / 255)
340
  )
341
- inpainted.save(f't{ix}_op.png')
342
  minpainted = mask_to_alpha(inpainted, m)
343
  # continue with partially inpainted image
344
  # since the tiles overlap, the next tile will contain (possibly inpainted) parts of the previous tile
 
301
  )
302
  i.putalpha(morig)
303
  img = i
304
+ # img.save('0.png')
305
  assert img.width == img.height
306
  assert img.width > 512 and img.width < 512*2
307
 
 
319
 
320
  for ix, tc in enumerate(tile_coords(img, n=2)):
321
  i = img.crop(tc)
322
+ # i.save(f't{ix}.png')
323
  m = i.getchannel('A')
324
 
325
  """Run a single prediction on the model"""
 
338
  inpainted,
339
  1-(np.array(m) / 255)
340
  )
341
+ # inpainted.save(f't{ix}_op.png')
342
  minpainted = mask_to_alpha(inpainted, m)
343
  # continue with partially inpainted image
344
  # since the tiles overlap, the next tile will contain (possibly inpainted) parts of the previous tile