minor update

Signed-off-by: YunLiu <[email protected]>
Project-MONAI · Oct 15, 2024 · 0e1d13f · 0e1d13f
1 parent 6b8c638
commit 0e1d13f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/spleen_ct_segmentation_real_time/scripts/inference.py b/models/spleen_ct_segmentation_real_time/scripts/inference.py
@@ -40,15 +40,15 @@ class InferenceWorkflow(PythonicWorkflow):
         from monai.bundle import create_workflow
         from monai.transforms import LoadImaged
         
-        workflow = create_workflow("inference.InferenceWorkflow", dataset_dir="/workspace/Data/Task09_Spleen")
+        workflow = create_workflow("inference.InferenceWorkflow")
         workflow.initialize()
         input_loader = LoadImaged(keys="image")
         workflow.dataflow.update(input_loader({"image": "/workspace/Data/Task09_Spleen/imagesTr/spleen_46.nii.gz"}))
         workflow.run()
         
         # update dataflow
         workflow.dataflow.clear()
-        workflow.dataflow.update({"image": "/workspace/Data/Task09_Spleen/imagesTr/spleen_38.nii.gz"})
+        workflow.dataflow.update(input_loader({"image": "/workspace/Data/Task09_Spleen/imagesTr/spleen_38.nii.gz"}))
         workflow.run()
 
     """