Update app.py

app.py

@@ -119,7 +119,14 @@ def get_images(batch: th.Tensor):
     reshaped = scaled.permute(2, 0, 3, 1).reshape([batch.shape[2], -1, 3])
     return Image.fromarray(reshaped.numpy())
 
-
+def get_images(batch: th.Tensor):
+    """ Display a batch of images inline. """
+    scaled = ((batch + 1)*127.5).round().clamp(0,255).to(th.uint8).cpu()
+    reshaped = scaled.permute(2, 0, 3, 1).reshape([batch.shape[2], -1, 3])
+    image = Image.fromarray(reshaped.numpy())
+    image = image.resize((image.width // 2, image.height // 2), Image.ANTIALIAS)
+    return image
+    
 # Create a classifier-free guidance sampling function
 guidance_scale = 8.0
 
@@ -133,8 +140,7 @@ def model_fn(x_t, ts, **kwargs):
     eps = th.cat([half_eps, half_eps], dim=0)
     return th.cat([eps, rest], dim=1)
 
-def to_base64(pil_image, size=(64, 64)):
-    pil_image.thumbnail(size, Image.ANTIALIAS)
+def to_base64(pil_image):
     buffered = BytesIO()
     pil_image.save(buffered, format="JPEG")
     return base64.b64encode(buffered.getvalue())