ToPilImage

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from huggan.pytorch.pix2pix.modeling_pix2pix import GeneratorUNet
 from PIL import Image
 from torchvision.utils import save_image
-from torchvision.transforms import Compose, Resize, ToTensor, Normalize
+from torchvision.transforms import Compose, Resize, ToTensor, Normalize, ToPilImage
 from diffusers.utils import load_image, make_image_grid
 
 transform = Compose(
@@ -22,8 +22,8 @@ def greet(input):
 	pixel_values = transform(image).unsqueeze(0)
 	output = generator(pixel_values)
 
-	return output
+	return ToPilImage(output)
 
-iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="coord_", value="/18/73237/95677")], outputs=[gr.Image(type="pil", width=256, label="Output Image")])
+iface = gr.Interface(fn=greet, inputs=[gr.Textbox(label="coord_zxy", value="/18/73237/95677")], outputs=[gr.Image(type="pil", width=256, label="Output Image")])
 
 iface.launch()