Update README.md

README.md

@@ -132,10 +132,10 @@ By evaluating these factors, you can select a cloud service provider that aligns
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM,AutoTokenizer
 
-config = PeftConfig.from_pretrained("mohamedemam/Em2-llama-7b")
-base_model = AutoModelForCausalLM.from_pretrained("NousResearch/Llama-2-7b-hf")
-model = PeftModel.from_pretrained(base_model, "mohamedemam/Em2-llama-7b")
-tokenizer = AutoTokenizer.from_pretrained("mohamedemam/Em2-llama-7b", trust_remote_code=True)
+config = PeftConfig.from_pretrained("mohamedemam/Em2-bloomz-7b")
+base_model = AutoModelForCausalLM.from_pretrained("mohamedemam/Em2-bloomz-7b")
+model = PeftModel.from_pretrained(base_model, "mohamedemam/Em2-bloomz-7b")
+tokenizer = AutoTokenizer.from_pretrained("mohamedemam/Em2-bloomz-7b", trust_remote_code=True)
 pipe=MyPipeline(model,tokenizer)
 print(pipe(context,quetion,answer,generate=True,max_new_tokens=4, num_beams=2, do_sample=False,num_return_sequences=1))
 ```