Update app.py

app.py

@@ -117,6 +117,15 @@ js="""
     <script>console.log("BOOOOOOOOOOOOOOOOBS");</script>
 """
 
+js_upload = """
+() => {
+    const link = document.createElement('a');
+    link.href = output_data;
+    link.download = 'uploaded_image.png';
+    link.click();
+}
+"""
+
 with gr.Blocks(fill_width=True, head=js) as demo:
     with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
@@ -142,7 +151,7 @@ with gr.Blocks(fill_width=True, head=js) as demo:
                 with gr.Row():
                     output = [gr.Image(label=m, show_download_button=True, 
                               interactive=False, width=112, height=112, show_share_button=False, format="png",
-                              visible=True) for m in default_models]
+                              visible=True, js=js_upload) for m in default_models]
                     current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
         with gr.Column(scale=2):