Update README.md

README.md

@@ -6,4 +6,28 @@ tags:
 - CXR
 - CTPA
 - pulmonary-embolism
----
+---
+
+# Pulmonary Embolism Classifier 🧠
+
+This is a classification model for pulmonary embolism classification from CTPA scans. 
+This classification model is used to aid a classification task to a diffusion model that generated 3D CTPA scans from a single 2D chest X-ray.
+
+The input to the classification model are CTPA scan latents of size 32 x 32 x 64.
+This model was trained on CTPAs that were downscaled from size 256 x 256 x 64 in a 2D slice-by-slice manner using a VAE.
+
+This model is used in this git repository: https://github.com/NoaCahan/X-ray2CTPA
+
+## 🧩 Model Details
+- **Architecture:** 3D 121-DenseNet  
+- **Framework:** PyTorch  
+- **Input:** 3D volumes (64×32×32)  
+- **Output:** PE Binary Classification probability
+
+## 🧠 How to Use
+```python
+from huggingface_hub import hf_hub_download
+import torch
+
+model_path = hf_hub_download("username/medical-diffusion-classifier", "classification_model_256.pth.tar")
+model = torch.load(model_path, weights_only=False)