Update app.py

app.py

@@ -46,7 +46,7 @@ class Generator(nn.Module):
 
 
 @torch.no_grad()
-def interpolate(model, save_dir='./lerp/', frames=100, rows=8, cols=8):
+def interpolate(model, save_dir='./lerp/', frames=100, rows=1, cols=1):
     save_dir = Path(save_dir)
     save_dir.mkdir(exist_ok=True, parents=True)
 
@@ -89,7 +89,7 @@ def predict(model_name, choice, seed):
         interpolate(model)
         return 'out.gif'
     else:
-        z = torch.randn(64, 100, 1, 1)
+        z = torch.randn(1, 100, 1, 1)
         punks = model(z)
         save_image(punks, "image.png", normalize=True)
         img = Image.open(f"image.png").convert('RGBA')