Update app.py

app.py

@@ -24,7 +24,7 @@ def predict(images, eps):
   zca = K.enhance.ZCAWhitening(eps=eps, compute_inv=True)
   zca.fit(images)
   zca_images = zca(images)
-  grid_zca = make_grid(zca_images, nrow=2, normalize=True).cpu().numpy()
+  grid_zca = make_grid(zca_images, nrow=3, normalize=True).cpu().numpy()
   return np.transpose(grid_zca,[1,2,0])
 
 title = 'ZCA Whitening with Kornia!'
@@ -50,7 +50,7 @@ iface = gr.Interface(fn=predict,
                   'Carnation.jpg', 
                   'Orchid.jpg', 
                   'Peony.jpg'
-                  ], 0.01]]
+                  ], 0.1]]
              )
 
 if __name__ == "__main__":