AlekseyKorshuk commited on
Commit
112d8a8
1 Parent(s): 549eced

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ class Generator(nn.Module):
46
 
47
 
48
  @torch.no_grad()
49
- def interpolate(model, save_dir='./lerp/', frames=100, rows=1, cols=1):
50
  save_dir = Path(save_dir)
51
  save_dir.mkdir(exist_ok=True, parents=True)
52
 
@@ -89,7 +89,7 @@ def predict(model_name, choice, seed):
89
  interpolate(model)
90
  return 'out.gif'
91
  else:
92
- z = torch.randn(1, 100, 1, 1)
93
  punks = model(z)
94
  save_image(punks, "image.png", normalize=True)
95
  img = Image.open(f"image.png").convert('RGBA')
 
46
 
47
 
48
  @torch.no_grad()
49
+ def interpolate(model, save_dir='./lerp/', frames=100, rows=8, cols=8):
50
  save_dir = Path(save_dir)
51
  save_dir.mkdir(exist_ok=True, parents=True)
52
 
 
89
  interpolate(model)
90
  return 'out.gif'
91
  else:
92
+ z = torch.randn(64, 100, 1, 1)
93
  punks = model(z)
94
  save_image(punks, "image.png", normalize=True)
95
  img = Image.open(f"image.png").convert('RGBA')