Update app.py

app.py

@@ -85,7 +85,7 @@ def calculate_volume(mask_image_path):
 
     return size,spacing,V_Right_ventricular_cistern, V_Right_cerebral_sulcus, V_Left_ventricular_cistern, V_Left_cerebral_sulcus
     
-def process_nii_file(input_nii_file, dicom_file, slice, mode):
+def process_nii_file(input_nii_file, slice, mode):
             
     if mode == "Step1:Segment":
         device = torch.device("cuda" if torch.cuda.is_available() else "cpu")