Upload app.py

app.py

@@ -45,6 +45,7 @@ default_batch_size = 1
 
 
 def process_image_check(path_input):
+    logging.info(f"Input image path: {path_input}")
     if path_input is None:
         raise gr.Error(
             "Missing image in the first pane: upload a file or use one from the gallery below."
@@ -109,9 +110,6 @@ def run_demo_server(pipe, vae_2):
         theme=gradio_theme,
         title="Dereflection Any Image",
         css="""
-            #download {
-                height: 118px;
-            }
             .slider .inner {
                 width: 5px;
                 background: #FFF;
@@ -123,37 +121,15 @@ def run_demo_server(pipe, vae_2):
                 font-size: 20px !important;
                 color: crimson !important;
             }
-            h1 {
-                text-align: center;
-                display: block;
-            }
-            h2 {
-                text-align: center;
-                display: block;
-            }
-            h3 {
-                text-align: center;
-                display: block;
-            }
-            .md_feedback li {
-                margin-bottom: 0px !important;
-            }
-        """,
-        head="""
-            <script async src="https://www.googletagmanager.com/gtag/js?id=G-1FWSVCGZTG"></script>
-            <script>
-                window.dataLayer = window.dataLayer || [];
-                function gtag() {dataLayer.push(arguments);}
-                gtag('js', new Date());
-                gtag('config', 'G-1FWSVCGZTG');
-            </script>
         """,
     ) as demo:
         gr.Markdown(
             """
             # Dereflection Any Image
             <p align="center">
-        """
+            Upload an image to remove reflections.
+            </p>
+            """
         )
 
         with gr.Tabs(elem_classes=["tabs"]):
@@ -166,15 +142,14 @@ def run_demo_server(pipe, vae_2):
                         )
                         with gr.Row():
                             image_submit_btn = gr.Button(
-                                value="remove reflection", variant="primary"
+                                value="Remove Reflection", variant="primary"
                             )
                             image_reset_btn = gr.Button(value="Reset")
                     with gr.Column():
                         image_output_slider = ImageSlider(
-                            label="outputs",
+                            label="Outputs",
                             type="filepath",
                             show_download_button=True,
-                            show_share_button=True,
                             interactive=False,
                             elem_classes="slider",
                         )
@@ -211,7 +186,6 @@ def run_demo_server(pipe, vae_2):
             fn=lambda: (
                 None,
                 None,
-                None,
             ),
             inputs=[],
             outputs=[