Update README.md

README.md

@@ -18,6 +18,83 @@ datasets:
   - pdg-particle-data
 pipeline_tag: text-generation
 library_name: transformers
+model_size: 9B
+widget:
+  - text: "What is the invariant mass of two photons with energies 62.5 GeV each, traveling back-to-back?"
+    example_title: Invariant mass calculation
+  - text: "Explain the CMS detector architecture and its main subsystems."
+    example_title: Detector explanation
+  - text: "A Z boson decays at rest into an electron-positron pair. What is the electron momentum?"
+    example_title: Decay kinematics
+model-index:
+  - name: hep-agent-qwen-qwen3-5-9b-mi300x
+    results:
+      - task:
+          type: text-generation
+        dataset:
+          name: MMLU
+          type: cais/mmlu
+          config: all
+        metrics:
+          - name: MMLU (5-shot)
+            type: acc
+            value: 70.6
+            verified: false
+      - task:
+          type: text-generation
+        dataset:
+          name: ARC Challenge
+          type: allenai/ai2_arc
+          config: ARC-Challenge
+        metrics:
+          - name: ARC-Challenge (25-shot, norm)
+            type: acc_norm
+            value: 71.8
+            verified: false
+      - task:
+          type: text-generation
+        dataset:
+          name: MMLU Conceptual Physics
+          type: cais/mmlu
+          config: conceptual_physics
+        metrics:
+          - name: MMLU Conceptual Physics (5-shot)
+            type: acc
+            value: 77.9
+            verified: false
+      - task:
+          type: text-generation
+        dataset:
+          name: MMLU College Physics
+          type: cais/mmlu
+          config: college_physics
+        metrics:
+          - name: MMLU College Physics (5-shot)
+            type: acc
+            value: 58.8
+            verified: false
+      - task:
+          type: text-generation
+        dataset:
+          name: MMLU High School Physics
+          type: cais/mmlu
+          config: high_school_physics
+        metrics:
+          - name: MMLU High School Physics (5-shot)
+            type: acc
+            value: 62.9
+            verified: false
+      - task:
+          type: text-generation
+        dataset:
+          name: MMLU Astronomy
+          type: cais/mmlu
+          config: astronomy
+        metrics:
+          - name: MMLU Astronomy (5-shot)
+            type: acc
+            value: 80.9
+            verified: false
 ---
 
 # hep-agent-qwen-qwen3-5-9b-mi300x
@@ -28,8 +105,6 @@ This model is a full fine-tune of [Qwen/Qwen3.5-9B](https://huggingface.co/Qwen/
 corpus of High Energy Physics literature, experimental data, and synthetic Q&A.
 Trained on a single AMD MI300X (192 GB HBM3, ROCm 7.0).
 
-**Training date:** 2026-05-31  
-**Evaluation date:** 2026-05-31 (run `20260531_172915`)
 
 ## Model Overview
 
@@ -165,8 +240,19 @@ with torch.no_grad():
 print(tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))
 ```
 
+### Example 2
 
-### example 2
+```bash 
+
+# Install latest stable Transformers
+!pip install -U transformers==5.5.0
+
+# Install remaining deps
+!pip install -U accelerate bitsandbytes sentencepiece protobuf peft trl
+
+# Optional
+!pip install -U unsloth
+```
 
 ```python
 from transformers import (
@@ -174,6 +260,7 @@ from transformers import (
     AutoModelForCausalLM,
     BitsAndBytesConfig,
 )
+import torch
 
 model_name = "rajveer43/hep-agent-qwen-qwen3-5-9b-mi300x"
 
@@ -195,12 +282,11 @@ tokenizer = AutoTokenizer.from_pretrained(
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
     device_map="auto",
-    dtype=torch.float16,   
+    dtype=torch.float16,
     trust_remote_code=True,
     quantization_config=bnb_config,
 )
 
-import torch
 
 prompt = "Explain what a jet detector is in particle physics."
 
@@ -238,6 +324,7 @@ response = tokenizer.decode(
 print(response)
 ```
 
+
 ### vLLM Server (Recommended for Production)
 
 ```bash