|
|
--- |
|
|
license: mit |
|
|
datasets: |
|
|
- ai4bharat/indic-align |
|
|
language: |
|
|
- en |
|
|
metrics: |
|
|
- accuracy |
|
|
base_model: |
|
|
- microsoft/DialoGPT-small |
|
|
pipeline_tag: text-generation |
|
|
library_name: transformers |
|
|
tags: |
|
|
- text-generation-inference |
|
|
--- |
|
|
|
|
|
# anktechsol/ankiGPT-small |
|
|
|
|
|
## 🧠 What is ankiGPT-small? |
|
|
A conversational **text-generation** model fine-tuned from `microsoft/DialoGPT-small` for Indian scenarios—supporting English and Hinglish. Use it to generate stories, dialogue, quick responses, and creative text. |
|
|
|
|
|
--- |
|
|
|
|
|
## 🚀 Quick Start |
|
|
|
|
|
```python |
|
|
from transformers import pipeline |
|
|
generator = pipeline("text-generation", model="anktechsol/ankiGPT-small") |
|
|
prompt = "Write a short story about a day in the life of a student in a bustling Indian city." |
|
|
result = generator(prompt, max_length=300, num_return_sequences=1) |
|
|
print(result[0]['generated_text']) |
|
|
``` |
|
|
|
|
|
*Copy-paste this code to see instant results!* |
|
|
|
|
|
--- |
|
|
|
|
|
## ✨ Features |
|
|
|
|
|
- **Conversational:** Tuned for chat, stories, and messages |
|
|
- **Language:** English + Hinglish (Indian conversational flavor) |
|
|
- **Base Model:** DialoGPT-small |
|
|
- **Size:** 124M parameters (fast and lightweight) |
|
|
- **Dataset:** ai4bharat/indic-align (Indian context data) |
|
|
|
|
|
--- |
|
|
|
|
|
## 💡 Example Outputs |
|
|
|
|
|
> **Prompt:** "Describe the Diwali celebrations in Mumbai." |
|
|
> |
|
|
> **Output:** "The city sparkled with thousands of lights, families prepared delicious sweets, and friends gathered for bursting crackers, laughter echoing through the alleys." |
|
|
> |
|
|
> *Try your own prompts above!* |
|
|
|
|
|
--- |
|
|
|
|
|
## ⚠️ Limitations & Considerations |
|
|
|
|
|
- Tends to repeat on long text—adjust `max_length` and `no_repeat_ngram_size` as needed |
|
|
- Biased towards Indian contexts due to training data |
|
|
- Not for critical or factual information generation |
|
|
|
|
|
--- |
|
|
|
|
|
## 🙌 Contributions & Community |
|
|
|
|
|
- **Suggestions?** Open an issue or start a discussion. We welcome community feedback! |
|
|
- **Demo:** Want a hands-on demo? Let us know! |
|
|
|
|
|
--- |
|
|
|
|
|
## 🔗 References |
|
|
|
|
|
- [anktechsol.com](https://www.anktechsol.com) |
|
|
|
|
|
--- |
