save image

app.py

@@ -16,6 +16,7 @@ import numpy as np
 import torch
 import torch.nn.functional as F
 from torchvision.utils import save_image
+from torchvision.transforms import ToPILImage
 from huggingface_hub import hf_hub_download
 
 from absl import logging
@@ -230,10 +231,13 @@ def infer(
         else:
             image_unprocessed = decode(_z)
 
-        samples = unpreprocess(image_unprocessed).contiguous()[0]
+        samples = unpreprocess(image_unprocessed).contiguous()
 
-    # return samples, seed
-    return seed
+        to_pil = ToPILImage()
+        
+        pil_image = to_pil(sample[0])
+
+    return pil_image, seed
 
 
 # examples = [
@@ -310,7 +314,7 @@ with gr.Blocks(css=css) as demo:
                 )
             with gr.Row():
                 num_of_interpolation = gr.Slider(
-                    label="Number of images for interpolation",
+                    label="Number of images for interpolation - More images yield smoother transitions but require more resources and may fail.",
                     minimum=5,
                     maximum=50,
                     step=1,
@@ -330,8 +334,7 @@ with gr.Blocks(css=css) as demo:
             num_inference_steps,
             num_of_interpolation,
         ],
-        # outputs=[result, seed],
-        outputs=[seed],
+        outputs=[result, seed],
     )
 
 if __name__ == "__main__":