Update README.md

README.md

@@ -23,11 +23,14 @@ To start using the Rashik24/Mistral-Instruct-Bangla model, you can use the follo
 ```Python
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
+from peft import PeftModel, PeftConfig
+
+
 
 def load_model(model_name):
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name)
-    return model, tokenizer
+    config = PeftConfig.from_pretrained("Rashik24/Mistral-Instruct-Bangla")
+    model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
+    model = PeftModel.from_pretrained(model, "Rashik24/Mistral-Instruct-Bangla")
 
 def generate_text(prompt, model, tokenizer):
     inputs = tokenizer.encode(prompt, return_tensors='pt')