File size: 949 Bytes
f211e05 0e40916 dfe7a9a 0e40916 c057496 6dd4b2d c057496 1c299fa c057496 6dd4b2d 8e69c0d 6a29910 dec5886 1c299fa dec5886 6a29910 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
language:
- ar
metrics:
- accuracy
- bleu
library_name: transformers
pipeline_tag: text2text-generation
---
This model is under trial.
The number in the generated text represents the category of the news, as shown below.
category_mapping = {
'Political':1,
'Economy':2,
'Health':3,
'Sport':4,
'Culture':5,
'Technology':6,
'Art':7,
'Accidents':8
}
# Example usage
model_name = "Hezam/arabic-T5-news-classification-generation"
from transformers import T5ForConditionalGeneration, T5Tokenizer
model = T5ForConditionalGeneration.from_pretrained(model_name)
tokenizer = T5Tokenizer.from_pretrained(model_name)
input_text = " الاستاذ حزام جوبح يحصل على براعة اختراع في التعلم العميق"
output_text = model.generate(input_text)
print(generated_text)
|