WiNE-iNEFF commited on
Commit
11c0506
·
1 Parent(s): 193c1a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ pipeline_name = 'WiNE-iNEFF/Minecraft-Skin-Diffusion-V2'
21
  image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
22
 
23
  def show_images_save(x):
24
- grid = torchvision.utils.make_grid(x, nrow=4)
25
  grid_im = grid.detach().cpu().permute(1, 2, 0).clip(0, 1) * 255
26
  grid_im = Image.fromarray(np.array(grid_im).astype(np.uint8))
27
  return grid_im
 
21
  image_pipe = DDPMPipeline.from_pretrained(pipeline_name).to(device)
22
 
23
  def show_images_save(x):
24
+ grid = torchvision.utils.make_grid(x, nrow=4, padding=0)
25
  grid_im = grid.detach().cpu().permute(1, 2, 0).clip(0, 1) * 255
26
  grid_im = Image.fromarray(np.array(grid_im).astype(np.uint8))
27
  return grid_im