malmarjeh commited on
Commit
c335573
·
1 Parent(s): bd6fe6e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +47 -0
README.md ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - ar
4
+ tags:
5
+ - AraGPT2
6
+ - GPT-2
7
+ - MSA
8
+ - Arabic Text Summarization
9
+ - Arabic News Title Generation
10
+ - Arabic Paraphrasing
11
+ widget:
12
+ - text: "شهدت مدينة طرابلس، مساء أمس الأربعاء، احتجاجات شعبية وأعمال شغب لليوم الثالث على التوالي، وذلك بسبب تردي الوضع المعيشي والاقتصادي. واندلعت مواجهات عنيفة وعمليات كر وفر ما بين الجيش اللبناني والمحتجين استمرت لساعات، إثر محاولة فتح الطرقات المقطوعة، ما أدى إلى إصابة العشرات من الطرفين."
13
+ ---
14
+
15
+ # An Arabic abstractive text summarization model
16
+ A fine-tuned AraGPT2 model on a dataset of 84,764 paragraph-summary pairs.
17
+
18
+ More details on the fine-tuning of this model will be released later.
19
+
20
+ The model can be used as follows:
21
+ ```python
22
+ from transformers import GPT2TokenizerFast, AutoModelForCausalLM
23
+ from arabert.preprocess import ArabertPreprocessor
24
+
25
+ model_name="malmarjeh/gpt2"
26
+ preprocessor = ArabertPreprocessor(model_name="")
27
+
28
+ tokenizer = GPT2TokenizerFast.from_pretrained("aubmindlab/aragpt2-base")
29
+ model = AutoModelForCausalLM.from_pretrained(model_name)
30
+
31
+ text = "شهدت مدينة طرابلس، مساء أمس الأربعاء، احتجاجات شعبية وأعمال شغب لليوم الثالث على التوالي، وذلك بسبب تردي الوضع المعيشي والاقتصادي. واندلعت مواجهات عنيفة وعمليات كر وفر ما بين الجيش اللبناني والمحتجين استمرت لساعات، إثر محاولة فتح الطرقات المقطوعة، ما أدى إلى إصابة العشرات من الطرفين."
32
+ text = preprocessor.preprocess(text)
33
+
34
+ text = '\n النص: ' + text + ' \n الملخص: \n '
35
+ tokenizer.add_special_tokens({'pad_token': '<pad>'})
36
+ tokens = tokenizer.batch_encode_plus([text], return_tensors='pt', padding='max_length', max_length=150)
37
+ output = model.generate(input_ids=tokens['input_ids'],repetition_penalty=3.0, num_beams=3, max_length=240, pad_token_id=2, eos_token_id=0, bos_token_id=10611)
38
+ result = tokenizer.decode(output[0][150:], skip_special_tokens=True).strip()
39
+ result
40
+ >>> 'واحتجاجات في طرابلس لليوم الثالث على التوالي'
41
+ ```
42
+
43
+ ## Contact:
44
+ **Mohammad Bani Almarjeh**: [Linkedin](https://www.linkedin.com/in/mohammad-bani-almarjeh/) | <[email protected]>
45
+
46
+
47
+