Focus README on model loading

Prioritize from_pretrained usage, keep the clone demo secondary, and link to the full project code and paper.

README.md

@@ -12,71 +12,14 @@ tags:
 
 # Seismic-LDDPM
 
-Inference-only Hugging Face release for seismic impedance inversion. The model
-weights are hosted in this repository in Diffusers format:
-
-```text
-vq_model/
-unet/
-scheduler/
-condition_encoder/
-```
-
-The repository also includes a minimal Overthrust inference script. Training
-code and experiment configs are not included here.
-
-## Quick Start
-
-```bash
-git clone https://huggingface.co/mally-2000/seismic-lddpm
-cd seismic-lddpm
-pip install -r requirements.txt
-```
-
-Run SAII-LDDPM:
-
-```bash
-python infer.py
-```
-
-Run SAII-CLDM:
-
-```bash
-python infer.py CLDM
-```
-
-`infer.py` builds the bundled Overthrust test sample, runs inference, and also
-runs the fixed Overthrust evaluation by default. Outputs are written under:
-
-```text
-outputs/infer_LDDPM/
-outputs/infer_CLDM/
-```
-
-## Results
-
-Local Overthrust impedance-domain results:
-
-| Method | Steps | PSNR | SSIM | PCC | RRE |
-|---|---:|---:|---:|---:|---:|
-| SAII-LDDPM | 1000 | 33.4413 | 0.9554 | 0.9957 | 0.0324 |
-| SAII-CLDM | 30 | 33.1312 | 0.9494 | 0.9950 | 0.0342 |
+Open model weights for seismic impedance inversion in Diffusers format. This
+repository is for inference and model loading; the full project code is at
+[GeoAI-INV/SAII-CLDM](https://github.com/GeoAI-INV/SAII-CLDM), and the paper is
+available on [arXiv](https://arxiv.org/html/2506.13529v1).
 
-## Python Loading
+## Load From Hugging Face
 
-Clone the repository and load local weights:
-
-```python
-import torch
-from codes.pipeline import SeismicImpInvLDDPMPipeline
-
-pipe = SeismicImpInvLDDPMPipeline.from_pretrained(
-    ".",
-    torch_dtype=torch.float32,
-).to("cuda")
-```
-
-Or let Diffusers download the model weights and custom pipeline code:
+SAII-LDDPM:
 
 ```python
 import torch
@@ -90,7 +33,7 @@ pipe = DiffusionPipeline.from_pretrained(
 ).to("cuda")
 ```
 
-For SAII-CLDM remote loading:
+SAII-CLDM:
 
 ```python
 pipe = DiffusionPipeline.from_pretrained(
@@ -100,3 +43,30 @@ pipe = DiffusionPipeline.from_pretrained(
     trust_remote_code=True,
 ).to("cuda")
 ```
+
+The inference algorithms are implemented in `codes/pipeline.py`. The root
+`pipeline.py` and `pipeline_cldm.py` files are thin entry points used by
+Diffusers remote loading.
+
+## Run The Bundled Demo
+
+```bash
+git clone https://huggingface.co/mally-2000/seismic-lddpm
+cd seismic-lddpm
+pip install -r requirements.txt
+
+python infer.py       # SAII-LDDPM
+python infer.py CLDM  # SAII-CLDM
+```
+
+`infer.py` uses the bundled Overthrust sample and writes outputs under
+`outputs/infer_LDDPM/` or `outputs/infer_CLDM/`.
+
+## Overthrust Results
+
+Impedance-domain metrics on the bundled Overthrust setting:
+
+| Method | Steps | PSNR | SSIM | PCC | RRE |
+|---|---:|---:|---:|---:|---:|
+| SAII-LDDPM | 1000 | 33.4413 | 0.9554 | 0.9957 | 0.0324 |
+| SAII-CLDM | 30 | 33.1312 | 0.9494 | 0.9950 | 0.0342 |