Update README.md

README.md

@@ -1,8 +1,6 @@
 ---
 license: mit
 base_model: facebook/bart-large-cnn
-tags:
-- generated_from_trainer
 model-index:
 - name: lora-bart-cnn-tib-1024
   results: []
@@ -44,6 +42,24 @@ Artificial Intelligence (AI) has become ingrained in our daily lives, from voice
 
 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