From 0e1d13f20c04416bb571cbd2e05b1d6326a5a914 Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Tue, 15 Oct 2024 22:34:00 +0800
Subject: [PATCH] minor update

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
---
 models/spleen_ct_segmentation_real_time/scripts/inference.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/models/spleen_ct_segmentation_real_time/scripts/inference.py b/models/spleen_ct_segmentation_real_time/scripts/inference.py
index 2fc76f79..e00b3d7f 100644
--- a/models/spleen_ct_segmentation_real_time/scripts/inference.py
+++ b/models/spleen_ct_segmentation_real_time/scripts/inference.py
@@ -40,7 +40,7 @@ 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"}))
@@ -48,7 +48,7 @@ class InferenceWorkflow(PythonicWorkflow):
         
         # 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()
 
     """