e

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, ToPilImage
+from torchvision.transforms import Compose, Resize, ToTensor, Normalize, ToPILImage
 from diffusers.utils import load_image, make_image_grid
 
 transform = Compose(
@@ -22,7 +22,7 @@ def greet(input):
 	pixel_values = transform(image).unsqueeze(0)
 	output = generator(pixel_values)
 
-	return ToPilImage(output)
+	return ToPILImage(output)
 
 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")])