Update main.py

main.py

@@ -34,5 +34,6 @@ def predict(mask):
     print(output.shape)
     return output[0, :, :, 0], output[0, :, :, 1], output[0, :, :, 2]
 
-demo = gradio.Interface(fn=predict, inputs=[gradio.Image(image_mode="L", source="canvas", tool="sketch")], outputs=[gradio.Image(image_mode="L"), gradio.Image(image_mode="L"), gradio.Image(image_mode="L")])
+demo = gradio.Interface(fn=predict, inputs=[gradio.Image(image_mode="L", source="canvas")], outputs=[gradio.Image(image_mode="L"), gradio.Image(image_mode="L"), gradio.Image(image_mode="L")])
+
 demo.run()