Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+from PIL import Image
 os.system("pip install gdown lpips gradio")
 os.system("gdown --id 1HKmjg6iXsWr4aFPuU0gBXPGR83wqMzq7 -O align.dat")
 os.system("wget https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/ffhq.pkl")
@@ -9,8 +10,6 @@ os.makedirs("embeddings/",exist_ok=True)
 os.makedirs("checkpoints/",exist_ok=True)
 os.makedirs("output/", exist_ok=True)
 os.makedirs("images/",exist_ok=True)
-
-
 import gradio as gr
 
 
@@ -18,9 +17,9 @@ import gradio as gr
 def inference(img):
     img.save("images/file.png")
     os.system("python tune.py")
-    return 
+    return Image.open("output/out.png")
   
-title = "Pivotal Tuning for Latent Based Real Image Editing"
-description = "Gradio Demo for Pivotal Tuning Inversion. 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."
-article = "<p style='text-align: center'><a href='https://github.com/danielroich/PTI' target='_blank'>Github Repo Pytorch</a>"
+title = "Pivotal Tuning for Latent-based editing of Real Images"
+description = "Gradio Demo for Pivotal Tuning Inversion. To use it, simply upload your image. Please use a cropped portrait picture for best results similar to the examples below."
+article = "<p style='text-align: center'><a href='https://github.com/danielroich/PTI' target='_blank'>Github Repo</a>"
 gr.Interface(inference, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="pil"),title=title,description=description,article=article,allow_flagging='never',allow_screenshot=False).launch(share=True)