Update app.py

app.py

@@ -493,7 +493,7 @@ def sample(model, image_size, batch_size=16, channels=3):
     return p_sample_loop(model, shape=(batch_size, channels, image_size, image_size))
 
 # sample 64 images
-sample_size = 25
+sample_size = 21
 samples = sample(best_model, image_size=image_size, batch_size=sample_size, channels=channels)
 
 """UI"""