Update main.py

main.py

@@ -72,6 +72,7 @@ with gradio.Blocks() as demo:
     gradio.Markdown("It can be challenging to rapidly infer the stress and strain that are present in a material with a complex microstructure. This demo runs a rapid surrogate model to compute strain for the microstructure that you draw!")     
 
     mask = gradio.Image(image_mode="L", source="canvas", label="microstructure")    
+    mask.update(fn=predict, inputs=[mask], outputs=[exx, eyy, exy, legend])
 
     btn = gradio.Button("Run!", variant="primary")
     exx = gradio.Image(label="ε-xx")