Update README.md

README.md

@@ -1,3 +1,4 @@
+```markdown
 ---
 license: apache-2.0
 datasets:
@@ -53,8 +54,10 @@ Ensure you have PyTorch and Hugging Face `transformers` installed:
 pip install torch transformers
 
 ```
-2. Inference Code
-Python
+
+### 2. Inference Code
+
+```python
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
@@ -85,4 +88,37 @@ outputs = model.generate(
 
 story = tokenizer.decode(outputs[0], skip_special_tokens=True)
 print("\n--- Generated Story ---")
-print(story)
+print(story)
+
+```
+
+---
+
+## Training Setup
+
+| Parameter | Configuration |
+| --- | --- |
+| **Hardware** | 1x NVIDIA T4 GPU (Google Colab Free Tier) |
+| **Training Duration** | ~1 hour |
+| **Dataset** | `roneneldan/TinyStories` |
+| **Framework** | PyTorch + Hugging Face Transformers |
+
+---
+
+## Citation & Acknowledgments
+
+If you use or reference **Pragya Preview**, please credit the underlying dataset:
+
+```bibtex
+@article{eldan2023tinystories,
+  title={TinyStories: How Small Can Language Models Be and Still Speak Coherent English?},
+  author={Eldan, Ronen and Li, Yuanzhi},
+  journal={arXiv preprint arXiv:2305.07759},
+  year={2023}
+}
+
+```
+
+```
+
+```