Update app.py

app.py

@@ -15,10 +15,9 @@ transform = Compose(
 model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes')
 
 def predict_fn(img):
-    img = Image.fromarray(img.astype('uint8'), 'RGB')
     inp = transform(img).unsqueeze(0)
     out = model(inp)
     save_image(out, 'out.png', normalize=True)
     return 'out.png'
 
-gr.Interface(predict_fn, inputs="sketchpad", outputs='image', examples=[['sample.jpg']]).launch()
+gr.Interface(predict_fn, inputs=gr.inputs.Image(type='pil'), outputs='image', examples=[['sample.jpg']]).launch()