Update app.py

app.py

@@ -20,6 +20,7 @@ from model import *
 from e4e_projection import projection as e4e_projection
 
 from copy import deepcopy
+import imageio
 
 os.makedirs('inversion_codes', exist_ok=True)
 os.makedirs('style_images', exist_ok=True)
@@ -82,8 +83,8 @@ def inference(img):
         original_my_sample = original_generator(my_w, input_is_latent=True)
         my_sample = generator(my_w, input_is_latent=True)   
     npimage = my_sample[0].permute(1, 2, 0).detach().numpy()[:,:,::-1] 
-    convert = (npimage * 255).astype('uint8')
-    return convert
+    imageio.imwrite('filename.jpeg', npimage)
+    return 'filename.jpeg'
   
 title = "JojoGAN"
 description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, simply upload your image, or click one of the examples to load them. Read more at the links below."
@@ -91,4 +92,4 @@ description = "Gradio Demo for JoJoGAN: One Shot Face Stylization. To use it, si
 article = "<p style='text-align: center'><a href='https://github.com/mchong6/JoJoGAN' target='_blank'>Github Repo Pytorch</a></p> <center><img src='https://visitor-badge.glitch.me/badge?page_id=akhaliq_jojogan' alt='visitor badge'></center> <p style='text-align: center'>samples from repo: <img src='https://raw.githubusercontent.com/mchong6/JoJoGAN/main/teaser.jpg' alt='animation'/></p>"
 
 examples=[['iu.jpeg']]
-gr.Interface(inference, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="numpy"),title=title,description=description,article=article,enable_queue=True,allow_flagging=False,examples=examples).launch()
+gr.Interface(inference, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="file"),title=title,description=description,article=article,enable_queue=True,allow_flagging=False,examples=examples).launch()