Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("RockySinghRajput/Indra", torch_dtype="auto", device_map="auto")
4tokenizer = AutoTokenizer.from_pretrained("RockySinghRajput/Indra")
5
6messages = [
7 {"role": "system", "content": "आप Indra हैं, एक बुद्धिमान AI सहायक।"},
8 {"role": "user", "content": "भारत के स्वतंत्रता संग्राम के बारे में बताइए।"}
9]
10
11text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
12inputs = tokenizer([text], return_tensors="pt").to(model.device)
13
14output = model.generate(inputs.input_ids, max_new_tokens=512, temperature=0.7, top_p=0.9)
15print(tokenizer.decode(output[0][len(inputs.input_ids[0]):], skip_special_tokens=True))1messages = [
2 {"role": "system", "content": "You are Indra, a helpful AI that understands Hindi and English."},
3 {"role": "user", "content": "Mujhe Python mein ek calculator banana hai, kaise karoon?"}
4]| Property | Value |
|---|---|
| Base Model | Qwen/Qwen2.5-Coder-1.5B-Instruct |
| Parameters | 1.5B |
| Type | Causal Language Model (merged LoRA fine-tune) |
| Languages | English, Hindi, Bengali, Tamil, Telugu, Marathi, Gujarati, Kannada, Malayalam, Punjabi, Odia |
| License | Apache 2.0 |
| Developed by | RockySinghRajput |
| Related Model | IndraCoder (coding-focused version) |
| Language | Script | Code |
|---|---|---|
| Hindi | देवनागरी | hi |
| Bengali | বাংলা | bn |
| Tamil | தமிழ் | ta |
| Telugu | తెలుగు | te |
| Marathi | मराठी | mr |
| Gujarati | ગુજરાતી | gu |
| Kannada | ಕನ್ನಡ | kn |
| Malayalam | മലയാളം | ml |
| Punjabi | ਪੰਜਾਬੀ | pa |
| Odia | ଓଡ଼ିଆ | or |
| English | Latin | en |
| Dataset | Purpose | Content |
|---|---|---|
| CohereForAI/aya_dataset | Multilingual Indian conversations | 10 Indian languages |
| Cognitive-Lab/Hindi-Instruct | Hindi instruction following | Hindi Q&A |
| sarvamai/samvaad-hi-en-instruct-v2 | Bilingual conversations | Hindi-English |
| CohereForAI/aya_collection (India-filtered) | Indian history & culture | History, heritage, knowledge |
| CohereForAI/aya_collection (Hindi WikiQA) | Hindi knowledge base | Wikipedia-sourced Hindi QA |
| ai4bharat/IndicSentiment | Hindi language understanding | Sentiment analysis |
| Test | Language | Task | Result |
|---|---|---|---|
| Hindi Chat | Hindi | Gandhi's role in freedom struggle | ✅ Detailed response |
| Indian History | English | Gupta Empire Golden Age | ✅ Accurate overview |
| Hinglish | Mixed | "Python mein calculator banana hai" | ✅ Code + Hindi explanation |
| Coding | English | Binary search implementation | ✅ Working code |
| Indian Culture | Hindi | Classical music ragas | ✅ Cultural knowledge |
| Model | Focus | Repo |
|---|---|---|
| Indra (this model) | Indian languages + history | RockySinghRajput/Indra |
| IndraCoder | Coding + debugging | RockySinghRajput/Indracoder |
| IndraCoder-7B | Advanced coding + chat (coming soon) | RockySinghRajput/IndraCoder-7B |
1@misc{indra2025,
2 title={Indra: An Indian Language AI Assistant},
3 author={RockySinghRajput},
4 year={2025},
5 publisher={HuggingFace},
6 url={https://huggingface.co/RockySinghRajput/Indra}
7}