Update README.md

README.md

@@ -11,7 +11,7 @@ A Vision Transformer model trained on multi-physics simulation data for scientif
 - **Developed by:** PhysicsAlchemists Research Team
 - **Model type:** Vision Transformer (ViT-Huge)
 - **Language(s):** N/A (Computer Vision)
-- **License:** Apache 2.0
+- **License:** MIT Licence
 - **Finetuned from model:** Trained from scratch on physics simulation data
 - **Training Steps:** 195,930 steps
 
@@ -70,13 +70,13 @@ from PIL import Image
 import torch
 
 # Load model and processor
-model = AutoModel.from_pretrained("your-username/physics-vit-full")
-processor = AutoImageProcessor.from_pretrained("your-username/physics-vit-full")
+model = AutoModel.from_pretrained("JessicaE/physics-vit-full")
+processor = AutoImageProcessor.from_pretrained("JessicaE/physics-vit-full")
 
 # Load your physics image
 image = Image.open("physics_simulation.png").convert('RGB')
 
-# ⚠️ CRITICAL: Apply custom preprocessing
+# Apply custom preprocessing
 image = expand_to_square(image, background_color=(128, 128, 128))
 image = image.resize((224, 224), Image.BILINEAR)
 
@@ -167,12 +167,12 @@ pip install transformers torch torchvision pillow
 ## Citation
 
 ```bibtex
-@misc{physics-vit-2024,
-  title={Physics Foundation Vision Transformer for Scientific Computing},
-  author={PhysicsAlchemists Research Team},
-  year={2024},
+@misc{physics-vit-2025,
+  title={PhySiViT : A Physics Simulation Vision Transformer},
+  author={Jessica Ezemba, James Afful, Mei-Yu Wang},
+  year={2025},
   howpublished={HuggingFace Model Hub},
-  url={https://huggingface.co/your-username/physics-vit-full}
+  url={https://huggingface.co/JessicaE/physics-vit-full}
 }
 ```
 
@@ -181,3 +181,8 @@ pip install transformers torch torchvision pillow
 - Built using [Cerebras ModelZoo](https://github.com/Cerebras/modelzoo)
 - Trained on Cerebras CS-X systems
 - Based on Vision Transformer architecture
+- This work was made possible thanks to the ByteBoost cybertraining program which is funded by the National Science Foundation Cybertraining awards: 2320990, 2320991, and 2320992, and the Neocortex project, the ACES platform, and the Ookami cluster.
+- The Neocortex project is supported by National Science Foundation award number 2005597.
+- The ACES (Accelerating Computing for Emerging Sciences) platform was funded by National Science Foundation award number 2112356.
+- The Ookami cluster is supported by National Science Foundation award number 1927880.
+