Ahsen Khaliq commited on
Commit
e59cfb4
1 Parent(s): 46a015d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -84,8 +84,9 @@ def inference(img):
84
 
85
  original_my_sample = original_generator(my_w, input_is_latent=True)
86
  my_sample = generator(my_w, input_is_latent=True)
87
- npimage = my_sample[0].permute(1, 2, 0).detach().numpy()
88
- return npimage
 
89
 
90
  title = "AnimeGANv2"
91
  description = "Gradio Demo for AnimeGanv2 Face Portrait. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."
 
84
 
85
  original_my_sample = original_generator(my_w, input_is_latent=True)
86
  my_sample = generator(my_w, input_is_latent=True)
87
+ npimage = my_sample[0].permute(1, 2, 0).detach().numpy()
88
+ convert = (npimage * 255).astype('uint8')
89
+ return convert
90
 
91
  title = "AnimeGANv2"
92
  description = "Gradio Demo for AnimeGanv2 Face Portrait. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below. Please use a cropped portrait picture for best results similar to the examples below."