Update README.md

README.md

@@ -1,19 +1,20 @@
 ---
 library_name: transformers
-tags: []
+language:
+- en
+base_model: google/gemma-2-9b-it
 ---
 
 # Buddhi-indic
 
-## Model Description
-
-**Model ID**: aiplanet/buddhi-indic
 
-**Language(s)**: Hindi, Kannada, Tamil
 
-**Architecture**: Gemma2ForCausalLM
+## Model Description
 
-**Training Data**: This model has been fine-tuned on diverse datasets encompassing Hindi, Kannada, and Tamil languages to provide robust performance across these Indic languages.
+- **Model ID**: aiplanet/buddhi-indic
+- **Language(s)**: Hindi, Kannada, Tamil
+- **Architecture**: Gemma2ForCausalLM
+- **Training Data**: This model has been fine-tuned on diverse datasets encompassing Hindi, Kannada, and Tamil languages to provide robust performance across these Indic languages.
 
 ## Intended Use
 
@@ -40,7 +41,7 @@ prompt_template = """
 ```python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_name = 'aiplanet/Gemma-2-9b-indic'
+model_name = 'aiplanet/buddhi-indic'
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
@@ -57,7 +58,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 import torch
 
-model_name = 'aiplanet/Gemma-2-9b-indic'
+model_name = 'aiplanet/buddhi-indic'
 
 bnb_config = BitsAndBytesConfig(
     load_in_4bit=True,
@@ -158,4 +159,4 @@ tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
    बास्केटबॉल का आविष्कार जेम्स नेस्मिथ ने 1891 में किया था। ...
    ```
 
----
+---