Initial CosAE release

README.md

@@ -15,10 +15,16 @@ It encodes images into amplitude/phase harmonics and reconstructs RGB images.
 
 ## Usage
 ```python
-from cosae import CosAEModel
-model = CosAEModel.from_pretrained("vedant-jumle/cosae")
+from transformers import AutoModel
+
+# Load the model with remote code trust
+model = AutoModel.from_pretrained(
+    "vedant-jumle/cosae",
+    trust_remote_code=True,
+)
 model.eval()
-# Input should be a tensor of shape [B, 9, H, W] (RGB + FFT) or [B,3,H,W] for RGB-only.
+
+# Example input: tensor of shape [B, 9, H, W] (RGB + FFT) or [B,3,H,W]
 import torch
 x = torch.randn(1, 9, 256, 256)
 with torch.no_grad():