Spaces:

cmudrc
/

microstructure-strain

Build error

ccm commited on Nov 19, 2022

Commit

5405174

1 Parent(s): 9cc2dcb

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -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.run()

     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")], outputs=[gradio.Image(image_mode="L"), gradio.Image(image_mode="L"), gradio.Image(image_mode="L")])
 demo.run()