Update README.md

README.md

@@ -47,10 +47,10 @@ Try out the model with the code below :D
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-config = PeftConfig.from_pretrained("jolenechong/lora-bart-samsum-tib-1024")
-model = AutoModelForSeq2SeqLM.from_pretrained("philschmid/bart-large-cnn-samsum")
-model = PeftModel.from_pretrained(model, "jolenechong/lora-bart-samsum-tib-1024")
-tokenizer = AutoTokenizer.from_pretrained("jolenechong/lora-bart-samsum-tib-1024", from_pt=True)
+config = PeftConfig.from_pretrained("jolenechong/lora-bart-cnn-tib-1024")
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
+model = PeftModel.from_pretrained(model, "jolenechong/lora-bart-cnn-tib-1024")
+tokenizer = AutoTokenizer.from_pretrained("jolenechong/lora-bart-cnn-tib-1024", from_pt=True)
 
 text = """[add transcript you want to summarize here]"""
 inputs = tokenizer(text, return_tensors="pt")