Update README.md

README.md

@@ -21,22 +21,22 @@ tags:
 license: apache-2.0
 pipeline_tag: image-classification
 base_model:
-- galeio-research/nereus-sar-1
+- galeio-research/OceanSAR-1
 ---
 
-# Model Card for Nereus-SAR-1-SWH
+# Model Card for OceanSAR-1-wave
 
 ## Model Details
 
 ### Model Description
 
-Nereus-SAR-1-SWH is a linear probing head for significant wave height (SWH) prediction built on top of the Nereus-SAR-1 foundation model. It leverages the powerful features extracted by Nereus-SAR-1 to accurately predict ocean wave heights from Synthetic Aperture Radar (SAR) imagery.
+OceanSAR-1-wave is a linear probing head for significant wave height (SWH) prediction built on top of the OceanSAR-1 foundation model. It leverages the powerful features extracted by OceanSAR-1 to accurately predict ocean wave heights from Synthetic Aperture Radar (SAR) imagery.
 
 - **Developed by:** Thomas Kerdreux, Alexandre Tuel @ [Galeio](http://galeio.fr)
 - **Deployed by:** Antoine Audras @ [Galeio](http://galeio.fr)
 - **Model type:** Linear Regression Head on Vision Foundation Model
 - **License:** Apache License 2.0
-- **Base model:** Nereus-SAR-1 (ResNet50/ViT variants)
+- **Base model:** OceanSAR-1 (ResNet50/ViT variants)
 - **Training data:** Sentinel-1 Wave Mode (WV) SAR images with collocated wave height measurements
 
 ## Uses
@@ -69,7 +69,7 @@ import torch
 from transformers import AutoModelForImageClassification
 
 # Load the foundation model and the linear probing head
-nereus = AutoModelForImageClassification.from_pretrained("galeio-research/nereus-sar-1")
+oceansar = AutoModelForImageClassification.from_pretrained("galeio-research/OceanSAR-1")
 
 # Prepare your SAR image (should be single-channel VV polarization)
 # Here using random data as example
@@ -77,7 +77,7 @@ dummy_image = torch.randn(1, 1, 256, 256)  # (C, H, W)
 
 # Extract features
 with torch.no_grad():
-    outputs = nereus(dummy_image)
+    outputs = oceansar(dummy_image)
     # For regression, use the single output value as the wave height prediction
     wave_height = outputs.logits.item()  # Output in meters
 ```
@@ -88,7 +88,7 @@ with torch.no_grad():
 
 - **Dataset:** Sentinel-1 Wave Mode (WV) SAR images with collocated wave height measurements
 - **Source:** Wave height measurements from altimeters, buoys, and wave models
-- **Preprocessing:** Same as base Nereus-SAR-1 model
+- **Preprocessing:** Same as base OceanSAR-1 model
 
 ## Evaluation
 
@@ -119,11 +119,11 @@ The model outperforms existing approaches:
 
 - PyTorch >= 1.8.0
 - Transformers >= 4.30.0
-- Base Nereus-SAR-1 model
+- Base OceanSAR-1 model
 
 ### Input Specifications
 
-- Same as base Nereus-SAR-1 model
+- Same as base OceanSAR-1 model
 - Single channel (VV polarization) SAR images
 - 256x256 pixel resolution