From d60ed7e46b212a0284efc1d17525c19da96fbe19 Mon Sep 17 00:00:00 2001
From: Hadi Sotoudeh <mhs60@cam.ac.uk>
Date: Fri, 8 May 2026 14:47:02 +0100
Subject: [PATCH] updated readme to become consistent with the tutorial
 notebook

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 90237bf..aca5289 100644
--- a/README.md
+++ b/README.md
@@ -40,12 +40,12 @@ from aion.codecs import CodecManager
 from aion.modalities import LegacySurveyImage
 
 # Load model and codec manager
-model = AION.from_pretrained('aion-base').to('cuda')  # or 'aion-large', 'aion-xlarge'
+model = AION.from_pretrained('polymathic-ai/aion-base').to('cuda')  # or 'aion-large', 'aion-xlarge'
 codec_manager = CodecManager(device='cuda')
 
 # Prepare your astronomical data (example: Legacy Survey image)
 image = LegacySurveyImage(
-    flux=your_image_tensor,  # Shape: [batch, 4, height, width] for g,r,i,z bands
+    flux=your_image_torch_tensor,  # Shape: [batch, 4, height, width] for g,r,i,z bands
     bands=['DES-G', 'DES-R', 'DES-I', 'DES-Z']
 )