Update app.py

app.py

@@ -15,6 +15,8 @@ transform = Compose(
 model = GeneratorUNet.from_pretrained('huggan/pix2pix-uavid-15')
 
 def predict_fn(img):
+    img = Image.fromarray(np.array(img)[:, ::-1, :], "RGB")
+    img = img.convert("RGB")
     inp = transform(img).unsqueeze(0)
     out = model(inp)
     save_image(out, 'out.png', normalize=True)