Spaces:

andreped
/

livermask

Running

andreped commited on May 25, 2023

Commit

d3a6d05

•

2 Parent(s): a32d368 a222e79

Merge pull request #25 from andreped/2dviewer-fix

Files changed (1) hide show

demo/src/gui.py CHANGED Viewed

@@ -33,7 +33,7 @@ class WebUI:
         path = mesh_file_name.name
         run_model(path, model_name)
         nifti_to_glb("prediction-livermask.nii")
-        self.images = load_ct_to_numpy("./files/test_ct.nii")
         self.pred_images = load_pred_volume_to_numpy("./prediction-livermask.nii")
         self.slider = self.slider.update(value=2)
         return "./prediction.obj"

         path = mesh_file_name.name
         run_model(path, model_name)
         nifti_to_glb("prediction-livermask.nii")
+        self.images = load_ct_to_numpy(path)
         self.pred_images = load_pred_volume_to_numpy("./prediction-livermask.nii")
         self.slider = self.slider.update(value=2)
         return "./prediction.obj"