Ahsen Khaliq commited on
Commit
51513fc
1 Parent(s): 191f30c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def inference(img):
81
 
82
  original_my_sample = original_generator(my_w, input_is_latent=True)
83
  my_sample = generator(my_w, input_is_latent=True)
84
- npimage = my_sample[0].permute(1, 2, 0).detach().numpy()
85
  convert = (npimage * 255).astype('uint8')
86
  return convert
87
 
 
81
 
82
  original_my_sample = original_generator(my_w, input_is_latent=True)
83
  my_sample = generator(my_w, input_is_latent=True)
84
+ npimage = my_sample[0].permute(1, 2, 0).detach().numpy()[:,:,::-1]
85
  convert = (npimage * 255).astype('uint8')
86
  return convert
87