Update app.py

app.py

@@ -79,6 +79,7 @@ def run_one_image(img, model):
     show_image(im_paste[0], "reconstruction + visible")
 
     plt.show()
+    return plt
     
 
 # download checkpoint if not exist