Update app.py

app.py

@@ -33,9 +33,12 @@ 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)
-
-
-    with gr.Row():
+demo = gr.Blocks(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) 😛.
+                    """)
+with gr.Row():
         with gr.Column():
             with gr.Row():
                 Image_input = gr.Image(type="numpy", label="Input", interactive=True)