Update app.py

app.py

@@ -33,43 +33,47 @@ def switch_states(is_checked):
         return gr.Image.update(visible=True), gr.Button.update(visible=True)
     else:
         return gr.Image.update(visible=False), gr.Button.update(visible=False)
-demo = gr.Blocks(title="DISCO")
+
+demo = gr.Interface(title="DISCO")
 with demo:
     gr.Markdown(value="""
-                    **Gradio demo for DISCO: Disentangled Image Colorization via Global Anchors**. Check our [project page](https://menghanxia.github.io/projects/disco.html) 😛.
+                    Gradio demo for DISCO: Disentangled Image Colorization via Global Anchors. Check our [project page](https://menghanxia.github.io/projects/disco.html) 😛.
                     """)
-with gr.Row():
+
+    with gr.Row():
         with gr.Column():
             with gr.Row():
-                Image_input = gr.Image(type="numpy", label="Input", interactive=True)
-                Image_anchor = gr.Image(type="numpy", label="Anchor", tool="color-sketch", interactive=True, visible=False)
+                Image_input = gr.inputs.Image(type="numpy", label="Input", interactive=True)
+                Image_anchor = gr.inputs.Image(type="numpy", label="Anchor", tool="color-sketch", interactive=True, visible=False)
             with gr.Row():
-                Num_anchor = gr.Number(type="int", value=8, label="Num. of anchors (3~14)")
-                Radio_resolution = gr.Radio(type="index", choices=["Low (256x256)", "High (512x512)"], \
-                                                label="Colorization resolution (Low is more stable)", value="Low (256x256)")
+                Num_anchor = gr.inputs.Number(type="int", value=8, label="Num. of anchors (3~14)")
+                Radio_resolution = gr.inputs.Radio(choices=["Low (256x256)", "High (512x512)"], label="Colorization resolution (Low is more stable)", default="Low (256x256)")
             with gr.Row():
-                Ckeckbox_editable = gr.Checkbox(default=False, label='Show editable anchors')
-                Button_show_anchor = gr.Button(value="Predict anchors", visible=False)
-            Button_run = gr.Button(value="Colorize")
+                Checkbox_editable = gr.inputs.Checkbox(default=False, label='Show editable anchors')
+                Button_show_anchor = gr.inputs.Button(value="Predict anchors", visible=False)
+            Button_run = gr.inputs.Button(value="Colorize")
+
         with gr.Column():
-            Image_output = [gr.Image(type="numpy", label="Output").style(height=480), gr.Image(type="numpy", label="Output").style(height=480)]
+            Image_output = [gr.outputs.Image(type="numpy", label="Output").style(height=480), gr.outputs.Image(type="numpy", label="Output").style(height=480)]
+
+    output_image_anchor, output_button_show_anchor = gr.outputs.Image(), gr.outputs.Button()
+    output_image_output = [gr.outputs.Image(label="Output").style(height=480), gr.outputs.Image(label="Output").style(height=480)]
+
+    gr.Interface.Transforms(
+        output_image_anchor, output_button_show_anchor, func=switch_states, inputs=Checkbox_editable, outputs=[Image_anchor, Button_show_anchor]
+    )
+
+    gr.Interface.Button(Button_show_anchor, func=click_predanchors, inputs=[Image_input, Num_anchor, Radio_resolution, Checkbox_editable], outputs=Image_anchor)
+
+    gr.Interface.Button(Button_run, func=click_colorize, inputs=[Image_input, Image_anchor, Num_anchor, Radio_resolution, Checkbox_editable], outputs=output_image_output)
 
-    Ckeckbox_editable.change(fn=switch_states, inputs=Ckeckbox_editable, outputs=[Image_anchor, Button_show_anchor])
-    Button_show_anchor.click(fn=click_predanchors, inputs=[Image_input, Num_anchor, Radio_resolution, Ckeckbox_editable], outputs=Image_anchor)
-    Button_run.click(fn=click_colorize, inputs=[Image_input, Image_anchor, Num_anchor, Radio_resolution, Ckeckbox_editable], \
-                    outputs=Image_output)
-    
     ## guiline
-  
     if RUN_MODE != "local":
-        gr.Examples(examples=[
-                    ['04.jpg', 8, "Low (256x256)"],
-                   ], 
-                    inputs=[Image_input,Num_anchor,Radio_resolution], outputs=[Image_output], label="Используйте эту фотографию для примера кликнув по ней", cache_examples=False)
-   
+        gr.Interface.Examples(examples=[
+            ['04.jpg', 8, "Low (256x256)"],
+        ], inputs=[Image_input, Num_anchor, Radio_resolution], outputs=output_image_output, title="Используйте эту фотографию для примера кликнув по ней").initialize(auto_close=False, auto_close_timeout=999)
 
 if RUN_MODE == "local":
-    demo.launch(server_name='9.134.253.83',server_port=7788)
+    demo.launch(server_name="9.134.253.83", server_port=7788)
 else:
-    demo.queue(default_enabled=True, status_update_rate=5)
     demo.launch()