Update app.py

app.py

@@ -117,15 +117,6 @@ 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):
@@ -151,11 +142,11 @@ 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, js=js_upload) for m in default_models]
+                              visible=True) for m in default_models]
                     current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
         with gr.Column(scale=2):
-            gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="TheGallery",
+            gallery = gr.Gallery(label="Output", show_download_button=True,
                                 interactive=False, show_share_button=False, container=True, format="png",
                                 preview=True, object_fit="cover", columns=2, rows=2) 
 
@@ -195,12 +186,12 @@ with gr.Blocks(fill_width=True, head=js) as demo:
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    output2 = [gr.Image(label='', show_download_button=True, elem_classes="output",
+                    output2 = [gr.Image(label='', show_download_button=True,
                                interactive=False, width=112, height=112, visible=True, format="png",
                                show_share_button=False, show_label=False) for _ in range(max_images)]
 
         with gr.Column(scale=2):
-            gallery2 = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+            gallery2 = gr.Gallery(label="Output", show_download_button=True,
                                 interactive=False, show_share_button=True, container=True, format="png",
                                 preview=True, object_fit="cover", columns=2, rows=2)