Update README.md

README.md

@@ -18,4 +18,52 @@ cation, History, Language and Linguistics, Law,
 as well as Philosophy in Arabic.
 
 
-For more details, check out our [example.com](paper)
+For more details, check out our [example.com](paper)
+
+
+### Pipeline example
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+tokenizer = AutoTokenizer.from_pretrained("dru-acrps/ArGTClass")
+model = AutoModelForSequenceClassification.from_pretrained("dru-acrps/ArGTClass", device_map = 'auto')
+
+classifier = pipeline("text-classification", model=model, tokenizer= tokenizer)
+
+text = " .قصفت إسرائيل مستشفى المعمداني في مدينة غزة، والذي خلف مئات الشهداء والجرحى"
+
+classifier(text)
+
+```
+
+
+### Pipeline example (GPU)
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+tokenizer = AutoTokenizer.from_pretrained("dru-acrps/ArGTClass")
+model = AutoModelForSequenceClassification.from_pretrained("dru-acrps/ArGTClass", device_map = 'auto')
+
+classifier = pipeline("text-classification", model=model, tokenizer= tokenizer, device="cuda:0")
+
+text = " .قصفت إسرائيل مستشفى المعمداني في مدينة غزة، والذي خلف مئات الشهداء والجرحى"
+
+classifier(text)
+```
+
+### Full classification example
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+
+tokenizer = AutoTokenizer.from_pretrained("dru-acrps/ArGTClass")
+model = AutoModelForSequenceClassification.from_pretrained("dru-acrps/ArGTClass", device_map = 'auto')
+
+text = " .قصفت إسرائيل مستشفى المعمداني في مدينة غزة، والذي خلف مئات الشهداء والجرحى"
+
+inputs = tokenizer(text, return_tensors= 'pt')
+outputs = model(**inputs)
+ind = outputs.logits.argmax(dim=-1)[0]
+predicted_class = model.config.id2label[ind.item()]
+```
+