Update app.py

app.py

@@ -78,7 +78,8 @@ def run_one_image(img, model):
     plt.subplot(1, 4, 4)
     show_image(im_paste[0], "reconstruction + visible")
     
-    plt.savefig("test.png",bbox_inches='tight')
+    return plt
+    #plt.savefig("test.png",bbox_inches='tight')
     
 
 # download checkpoint if not exist
@@ -101,8 +102,8 @@ def inference(img):
   
   
   torch.manual_seed(2)
-  run_one_image(img, model_mae)
-  return "test.png"
+
+  return run_one_image(img, model_mae)
   
 
 title = "MAE"
@@ -111,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="file"),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="plot"),title=title,description=description,article=article,allow_flagging=False,allow_screenshot=False,enable_queue=True).launch()