Update README.md

README.md

@@ -30,6 +30,55 @@ It achieves the following results on the evaluation set:
 - Validation F1 Score (macro): 0.8105
 - Validation F1 Score (weighted): 0.8551
 
+## How to use
+
+You can use this model with a pipeline for text classification. 
+But first, you need to install the `peft` library like so: 
+
+```console 
+pip install peft
+```
+
+Then, you can run the following code.
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+
+model_id = "xlm-roberta-base"
+peft_model_id = "rasyosef/xlm-roberta-base-lora-amharic-news-classification" 
+
+categories = ['ሀገር አቀፍ ዜና', 'መዝናኛ', 'ስፖርት', 'ቢዝነስ', 'ዓለም አቀፍ ዜና', 'ፖለቲካ']
+id2label = {i: lbl for i, lbl in enumerate(categories)}
+label2id = {lbl: i for i, lbl in enumerate(categories)}
+
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+
+model = AutoModelForSequenceClassification.from_pretrained(
+    model_id, 
+    num_labels=len(categories), # 6
+    id2label=id2label,
+    label2id=label2id
+)
+
+model.load_adapter(peft_model_id)
+
+classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
+
+classifier([
+  """ቅርሶቹን ለመታደግ የተጀመረው የሙዚዬም ግንባታም በበጀት ምክንያት ተቋርጧል።
+  
+  በአፄ ቴዎድሮስ የንግስና ቦታ ደረስጌ ማሪያም ተጀምሮ የቆመው የሙዚየሙ ግንባታ ተጠናቀቆ ስራ 
+  እንዲጀምር ነዋሪዎች ጠይቀዋል። ዘመነ መሳፍንት መቋጫ ያገኘባት የኢትዮጵያ አንድነት የታወጀባት ዳግማዊ 
+  አፄ ቴዎድሮስ ከመንገሳቸው በፊት ደጃች ውቤን ቧሂት ከሚባል ቦታ ድል አድርገው ደጃች ውቤ ለንግስና ባዘጋጁት የንግስና ቦታና 
+  እቃዎች ንጉሰ ነገስት ዘኢትዮጵያ ተብለው የነገሱባት ቦታ ናት።""", # 'ሀገር አቀፍ ዜና'
+])
+```
+
+Output:
+```python
+[{'label': 'ሀገር አቀፍ ዜና', 'score': 0.977573037147522}]
+```
+
 ## Model Details
 
 ### Model Description