Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,7 @@ This model is used in this git repository: https://github.com/NoaCahan/X-ray2CTP
|
|
| 24 |
- **Input:** 3D volumes (64×32×32)
|
| 25 |
- **Output:** PE Binary Classification probability
|
| 26 |
|
|
|
|
| 27 |
## 🧠 How to Use
|
| 28 |
```python
|
| 29 |
from huggingface_hub import hf_hub_download
|
|
@@ -31,6 +32,7 @@ import torch
|
|
| 31 |
|
| 32 |
model_path = hf_hub_download("username/medical-diffusion-classifier", "classification_model_256.pth.tar")
|
| 33 |
model = torch.load(model_path, weights_only=False)
|
|
|
|
| 34 |
|
| 35 |
## 🧾 Citation
|
| 36 |
|
|
|
|
| 24 |
- **Input:** 3D volumes (64×32×32)
|
| 25 |
- **Output:** PE Binary Classification probability
|
| 26 |
|
| 27 |
+
|
| 28 |
## 🧠 How to Use
|
| 29 |
```python
|
| 30 |
from huggingface_hub import hf_hub_download
|
|
|
|
| 32 |
|
| 33 |
model_path = hf_hub_download("username/medical-diffusion-classifier", "classification_model_256.pth.tar")
|
| 34 |
model = torch.load(model_path, weights_only=False)
|
| 35 |
+
```
|
| 36 |
|
| 37 |
## 🧾 Citation
|
| 38 |
|