Added README info on how to run in-code | Minor bug fix with local path

CBICA · Jan 18, 2025 · 5b68174 · 5b68174
1 parent 413b2fa
commit 5b68174
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 1 deletion.
diff --git a/DLWMLS/__init__.py b/DLWMLS/__init__.py
@@ -1 +1 @@
-from dlwmls_pipeline import run_dlwmls_pipeline
+from DLWMLS.dlwmls_pipeline import run_dlwmls_pipeline
diff --git a/DLWMLS/__pycache__/__init__.cpython-312.pyc b/DLWMLS/__pycache__/__init__.cpython-312.pyc
diff --git a/DLWMLS/__pycache__/dlwmls_pipeline.cpython-312.pyc b/DLWMLS/__pycache__/dlwmls_pipeline.cpython-312.pyc
diff --git a/README.md b/README.md
@@ -31,6 +31,13 @@ A pre-trained nnUNet model can be found at our [HuggingFace account](https://hug
 ```bash
 DLWMLS -i "input_folder" -o "output_folder" -device cuda
 ```
+
+### In-code usage
+```python
+from DLWMLS import run_dlwmls_pipeline
+...
+run_dlwmls_pipeline(in_dir, out_dir, device)
+```
 For more details, please refer to
 
 ```bash
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from dlwmls_pipeline import run_dlwmls_pipeline
		from DLWMLS.dlwmls_pipeline import run_dlwmls_pipeline