Update app.py

app.py

@@ -1,18 +1,14 @@
 import gradio as gr
 import numpy as np
 
-def display_sketch(image):
-    return image
+def display_sketch(sketch):
+    # Directly return the sketch as it is already an image
+    return sketch
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Sketchpad Visualization")
-    with gr.Row():
-        with gr.Column():
-            sketchpad = gr.Sketchpad(label="Draw Something")
-        with gr.Column():
-            output_image = gr.Image(type="numpy", label="Your Sketch")
+    sketchpad = gr.Sketchpad(label="Draw Something")
+    output_image = gr.Image(label="Your Sketch")
 
-    sketchpad.change(fn=display_sketch, inputs=sketchpad, outputs=output_image)
+    sketchpad.submit(display_sketch, inputs=sketchpad, outputs=output_image)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch(share=True)