Update app.py

app.py

@@ -1,8 +1,14 @@
 import gradio as gr
+from PIL import Image
 import numpy as np
+import io
 
 def display_sketch(sketch):
-    # Directly return the sketch as it is already an image
+    # 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
 
 with gr.Blocks() as demo:
@@ -11,4 +17,4 @@ with gr.Blocks() as demo:
 
     sketchpad.change(display_sketch, inputs=sketchpad, outputs=output_image)
 
-demo.launch(share=True)
+demo.launch()