Downloads · 30 days
354
1% of all-time downloads
malmarjeh/t5-arabic-text-summarization
t5-arabic-text-summarization is a machine learning model from malmarjeh. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
A fine-tuned AraT5 model on a dataset of 84,764 paragraph-summary pairs.
Downloads · 30 days
354
1% of all-time downloads
All-time downloads
33.3K
Public
Repo size
4.2 GB
Likes
14
Public
Click a slice to open those files.
.bin1.1 GB · 100%
From the Hugging Face model README
A fine-tuned AraT5 model on a dataset of 84,764 paragraph-summary pairs.
Paper: Arabic abstractive text summarization using RNN-based and transformer-based architectures.
Dataset: link.
The model can be used as follows:
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
from arabert.preprocess import ArabertPreprocessor
model_name="malmarjeh/t5-arabic-text-summarization"
preprocessor = ArabertPreprocessor(model_name="")
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
pipeline = pipeline("text2text-generation",model=model,tokenizer=tokenizer)
text = "شهدت مدينة طرابلس، مساء أمس الأربعاء، احتجاجات شعبية وأعمال شغب لليوم الثالث على التوالي، وذلك بسبب تردي الوضع المعيشي والاقتصادي. واندلعت مواجهات عنيفة وعمليات كر وفر ما بين الجيش اللبناني والمحتجين استمرت لساعات، إثر محاولة فتح الطرقات المقطوعة، ما أدى إلى إصابة العشرات من الطرفين."
text = preprocessor.preprocess(text)
result = pipeline(text,
pad_token_id=tokenizer.eos_token_id,
num_beams=3,
repetition_penalty=3.0,
max_length=200,
length_penalty=1.0,
no_repeat_ngram_size = 3)[0]['generated_text']
result
>>> 'مواجهات عنيفة بين الجيش اللبناني ومحتجين في طرابلس'