Update README.md

README.md

@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # lora-bart-samsum-tib-1024
 
-This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on the TIB dataset.
+This model is a fine-tuned version of [philschmid/bart-large-cnn-samsum](https://huggingface.co/philschmid/bart-large-cnn-samsum) on the TIB dataset.
 
 ## Model description
 
@@ -42,6 +42,24 @@ Artificial Intelligence (AI) is a rapidly evolving technology that has profound
 
 Intended for summarizing video conferences/webinars.
 
+Try out the model with the code below :D
+```python
+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)
+
+text = """[add transcript you want to summarize here]"""
+inputs = tokenizer(text, return_tensors="pt")
+
+with torch.no_grad():
+    outputs = model.generate(input_ids=inputs["input_ids"])
+    print(tokenizer.batch_decode(outputs.detach().cpu().numpy())[0])
+```
+
 ## Training and evaluation data
 
 More information needed