Update app.py

app.py

@@ -1,20 +1,26 @@
 import gradio as gr
-from PIL import Image
-import numpy as np
-import io
 
 def display_sketch(sketch):
-    # Ensure the sketch is a numpy array
-    if isinstance(sketch, dict):
-        sketch = sketch["image"]
-    if not isinstance(sketch, np.ndarray):
-        raise ValueError("The sketch input is not a valid numpy array")
-    return sketch
+    import numpy as np
+    from PIL import Image
+    
+    image_data = sketch["image"]
+    img = Image.fromarray(np.uint8(image_data))
+    return img
 
-with gr.Blocks() as demo:
-    sketchpad = gr.Sketchpad(label="Draw Something")
-    output_image = gr.Image(label="Your Sketch")
+# Define the sketchpad component
+sketchpad = gr.Sketchpad(label="Draw Something")
 
-    sketchpad.change(display_sketch, inputs=sketchpad, outputs=output_image)
+# Define the image component to display the sketch
+output_image = gr.Image(label="Your Sketch")
 
+# Create the interface
+demo = gr.Interface(
+    fn=display_sketch, 
+    inputs=sketchpad, 
+    outputs=output_image,
+    live=True
+)
+
+# Launch the app
 demo.launch()