bhadresh-savani commited on
Commit
53b7a6c
·
1 Parent(s): 91cec6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -16,8 +16,6 @@ transform = Compose(
16
  model = GeneratorUNet.from_pretrained('huggan/pix2pix-uavid-15')
17
 
18
  def predict_fn(img):
19
- img = Image.fromarray(np.array(img)[:, ::-1, :], "RGB")
20
- img = img.convert("RGB")
21
  inp = transform(img).unsqueeze(0)
22
  out = model(inp)
23
  save_image(out, 'out.png', normalize=True)
 
16
  model = GeneratorUNet.from_pretrained('huggan/pix2pix-uavid-15')
17
 
18
  def predict_fn(img):
 
 
19
  inp = transform(img).unsqueeze(0)
20
  out = model(inp)
21
  save_image(out, 'out.png', normalize=True)