Update app.py

app.py

@@ -434,11 +434,18 @@ with gr.Blocks(theme=theme, css=css) as demo:
                 invert = gr.Checkbox(label="Invert", value=True)
                 resolution = gr.Slider(label="resolution", minimum=256, maximum=1024, value=512, step=64)
                 run_button = gr.Button("Lineart")
-            #            run_button = gr.Button(label="Run")
             with gr.Column():
                 gallery = gr.Gallery(label="Generated images", show_label=False, height="auto")
 
-        preprocessor_name.input(lambda preprocessor_name: gr.update(preprocessor_name), inputs=[preprocessor_name], outputs=[run_button])
+        def update_button_label(preprocessor_name):
+            if preprocessor_name == "Lineart":
+                return "Lineart"
+            elif preprocessor_name == "Lineart Coarse":
+                return "Lineart Coarse"
+            elif preprocessor_name == "Lineart Anime":
+                return "Lineart Anime"
+
+        preprocessor_name.change(fn=update_button_label, inputs=[preprocessor_name], outputs=[run_button])
         run_button.click(fn=lineart, inputs=[input_image, resolution, preprocessor_name, invert], outputs=[gallery])
 
     with gr.Tab("InPaint"):
@@ -446,13 +453,12 @@ with gr.Blocks(theme=theme, css=css) as demo:
             gr.Markdown("## InPaint")
         with gr.Row():
             with gr.Column():
-                #            input_image = gr.Image(source='upload', type="numpy", tool="sketch", height=512)
                 input_image = gr.ImageMask(sources="upload", type="numpy", height="auto")
                 invert = gr.Checkbox(label="Invert Mask", value=False)
                 run_button = gr.Button("Run")
-            #            run_button = gr.Button(label="Run")
+
             with gr.Column():
-                #            gallery = gr.Gallery(label="Generated images", show_label=False).style(height="auto")
+
                 gallery = gr.Gallery(label="Generated images", show_label=False, height="auto")
         run_button.click(fn=inpaint, inputs=[input_image, invert], outputs=[gallery])