Update README.md

README.md

@@ -23,18 +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):
-    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")
+    tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    return model, tokenizer
 
 def generate_text(prompt, model, tokenizer):
     inputs = tokenizer.encode(prompt, return_tensors='pt')
-    outputs = model.generate(inputs, max_length=50, num_return_sequences=1)
+    outputs = model.generate(inputs, max_length=256, num_return_sequences=1)
     return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 #Load the model