Update app.py

app.py

@@ -1,25 +1,16 @@
 import gradio as gr
-from PIL import Image
 import numpy as np
+from PIL import Image
 
 def display_sketch(sketch):
-    # Convert the sketch numpy array to an image
-    img = Image.fromarray(np.uint8(sketch))
-    return img
-
-# Define the sketchpad component
-sketchpad = gr.Sketchpad(label="Draw Something")
-
-# Define the image component to display the sketch
-output_image = gr.Image(label="Your Sketch")
+    # Convert the sketch (which is a numpy array) to a PIL image
+    image = Image.fromarray(sketch)
+    return image
 
-# Create the interface
-demo = gr.Interface(
-    fn=display_sketch, 
-    inputs=sketchpad, 
-    outputs=output_image,
-    live=True
-)
+with gr.Blocks() as demo:
+    sketchpad = gr.Sketchpad(label="Draw Something", shape=(256, 256))
+    output_image = gr.Image(label="Your Sketch")
+    
+    sketchpad.submit(display_sketch, inputs=sketchpad, outputs=output_image)
 
-# Launch the app
-demo.launch()
+demo.launch()