Update app.py

app.py

@@ -78,8 +78,7 @@ def run_one_image(img, model):
     plt.subplot(1, 4, 4)
     show_image(im_paste[0], "reconstruction + visible")
     
-    return plt
-    #plt.savefig("test.png",bbox_inches='tight')
+    plt.savefig("test.png",bbox_inches='tight')
     
 
 # download checkpoint if not exist
@@ -103,7 +102,8 @@ def inference(img):
   
   torch.manual_seed(2)
 
-  return run_one_image(img, model_mae)
+  run_one_image(img, model_mae)
+  return "test.png"
   
 
 title = "MAE"
@@ -112,4 +112,4 @@ description = "Gradio Demo for MAE. To use it, simply upload your image, or clic
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2112.11641' target='_blank'>JoJoGAN: One Shot Face Stylization</a>| <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>"
 
 
-gr.Interface(inference, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="plot"),title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False,enable_queue=True).launch()
+gr.Interface(inference, [gr.inputs.Image(type="pil")], gr.outputs.Image(type="file"),title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False,enable_queue=True).launch()