Update README.md

README.md

@@ -1,10 +1,72 @@
 ---
 title: README
 emoji: 🏃
-colorFrom: yellow
+colorFrom: indigo
 colorTo: gray
 sdk: static
 pinned: false
 ---
+# GEM_Testing_Arsenal 
 
-Edit this `README.md` markdown file to author your organization card.
+Welcome to ***GEM_Testing_Arsenal***, where groundbreaking research meets practical power! This repository unveils a novel architecture for On-Device Language Models (ODLMs), straight from our paper, ["Fragile Mastery: are domain-specific trade-offs undermining On-Device Language Models?"](./link_to_be_insterted). With just a few lines of code, our custom `gem_trainer.py` script lets you train ODLMs that are more accurate than ever, tracking accuracy and loss as you go.
+
+---
+
+### Highlights:
+- **Next-Level ODLMs**: Boosts accuracy with a new architecture from our research.
+- **Easy Training**: Call run_gem_pipeline to train on your dataset in minutes.
+- **Live Metrics**: Get accuracy and loss results as training unfolds.
+- **Flexible Design**: Works with any compatible dataset—plug and play!
+
+---
+### Prerequisites:
+To dive in, you’ll need:
+- **Python** `3.8+`
+
+- Required libraries (go through [quick start](#quick-start) below 👇)
+    
+- **Git** *(to clone the repo)*
+
+---
+### Quick Start:
+
+1. **Clone the repository:**
+    ```bash
+    git clone https://github.com/Firojpaudel/GEM.git
+    ```
+
+2. **Install Dependencies:**
+    ```pwsh
+    pip install -r requirements.txt
+    ```
+
+3. **Train Your Model:**
+Create a new python file and execute the code like:
+    ```python
+    from datasets import load_dataset
+    from gem_trainer import run_gem_pipeline
+
+    # Load a dataset (e.g., Banking77) {just replace the dataset here.}
+    dataset = load_dataset("banking77") 
+
+    # Train the ODLM
+    results = run_gem_pipeline(dataset, num_classes=77)
+
+    print(results)  # See accuracy and loss
+    ```
+
+> ***Boom—your ODLM is training with boosted accuracy!***
+
+---
+### Customizing Training:
+`run_gem_pipeline` keeps it simple, but you can tweak it! Dive into [`gem_trainer.py`](./gem_trainer.py) to adjust epochs, batch size, or other settings to fit your needs.
+
+---
+### Contributing 💓 
+Got ideas to make this even better? We’re all ears!
+- Fork the repo.
+- Branch off (`git checkout -b your-feature`).
+- Submit a pull request with your magic.
+
+---
+Edit this `README.md` markdown file to author your organization card.