Update app.py

app.py

@@ -35,7 +35,7 @@ def predict(mask):
   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", shape=(636, 101)), #, value=numpy.round(numpy.random.rand(101, 636, 3))), 
+                    inputs=gradio.Image(image_mode="L", source="canvas", tool="sketch", values=numpy.zeros(636, 101), #, value=numpy.round(numpy.random.rand(101, 636, 3))), 
                     outputs=[gradio.Image(image_mode="L"), gradio.Image(image_mode="L"), gradio.Image(image_mode="L")])
 
 if __name__ == "__main__":