Upload 2 files

demo.py

@@ -89,7 +89,7 @@ def process_nii_file(input_nii_file, dicom_file, slice, mode):
             
     if mode == "Step1:Segment":
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        root_dir = "./run"
+        root_dir = "./"
         model = UNETR(
             in_channels=1,
             out_channels=5,

requirements.txt

@@ -0,0 +1,7 @@
+matplotlib
+monai
+numpy
+torch
+gradio
+nibabel
+SimpleITK