Update README.md
Browse files
README.md
CHANGED
|
@@ -25,7 +25,7 @@ pip install open_clip_torch nibabel torch torchvision
|
|
| 25 |
|
| 26 |
|
| 27 |
##Load Pretrained Model
|
| 28 |
-
|
| 29 |
import torch
|
| 30 |
from model import BiomedClipClassifier, predict_from_paths
|
| 31 |
|
|
@@ -45,8 +45,8 @@ pred, probs = predict_from_paths(
|
|
| 45 |
print("Prediction:", pred)
|
| 46 |
print("Probabilities:", probs) # [CN, MCI, Dementia]
|
| 47 |
|
| 48 |
-
|
| 49 |
##Run Inference
|
| 50 |
-
|
| 51 |
python inference.py --weights . --mri /path/to/sample.nii.gz --text "Patient shows memory issues"
|
| 52 |
-
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
##Load Pretrained Model
|
| 28 |
+
```
|
| 29 |
import torch
|
| 30 |
from model import BiomedClipClassifier, predict_from_paths
|
| 31 |
|
|
|
|
| 45 |
print("Prediction:", pred)
|
| 46 |
print("Probabilities:", probs) # [CN, MCI, Dementia]
|
| 47 |
|
| 48 |
+
```
|
| 49 |
##Run Inference
|
| 50 |
+
```
|
| 51 |
python inference.py --weights . --mri /path/to/sample.nii.gz --text "Patient shows memory issues"
|
| 52 |
+
```
|