Update main.py

main.py

@@ -254,7 +254,8 @@ def predict(input_file):
     return (
         "./output/images/softmax-prostate-peripheral-zone-segmentation/prostate_gland_sm_pz.mha",
         "./output/images/softmax-prostate-central-gland-segmentation/prostate_gland_sm_tz.mha",
-        "./output/images/prostate-zonal-segmentation/prostate_gland.mha"
+        "./output/images/prostate-zonal-segmentation/prostate_gland.mha",
+    )
 
 print("Starting interface")
 demo = gr.Interface(