Update app.py

app.py

@@ -1,16 +1,12 @@
 import gradio as gr
-import numpy as np
-from PIL import Image
 
 def display_sketch(sketch):
-    # Convert the sketch (which is a numpy array) to a PIL image
-    image = Image.fromarray(sketch)
-    return image
+    return sketch
 
 with gr.Blocks() as demo:
-    sketchpad = gr.Sketchpad(label="Draw Something", shape=(256, 256))
+    sketchpad = gr.Sketchpad(label="Draw Something")
     output_image = gr.Image(label="Your Sketch")
-    
+
     sketchpad.submit(display_sketch, inputs=sketchpad, outputs=output_image)
 
 demo.launch()