Update app.py

app.py

@@ -178,7 +178,7 @@ def p_sample_loop(model, shape):
 
 @torch.no_grad()
 def sample(model, image_size, batch_size=16, channels=3):
-    return p_sample_loop(model, shape=(batch_size, channels, image_size, image_size))
+    return p_sample_loop(model, shape=(batch_size, channels, 64, 64))