Update README.md
Browse files
README.md
CHANGED
@@ -22,14 +22,10 @@ category_mapping = {
|
|
22 |
'Accidents':8
|
23 |
}
|
24 |
|
25 |
-
|
26 |
model_name = "Hezam/arabic-T5-news-classification-generation"
|
27 |
-
|
28 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
29 |
-
|
30 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
31 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
32 |
-
|
33 |
# Example usage
|
34 |
input_text = " الاستاذ حزام جوبح يحصل على براعة اختراع في التعلم العميق"
|
35 |
output_text = model.generate(input_text)
|
|
|
22 |
'Accidents':8
|
23 |
}
|
24 |
|
|
|
25 |
model_name = "Hezam/arabic-T5-news-classification-generation"
|
|
|
26 |
from transformers import T5ForConditionalGeneration, T5Tokenizer
|
|
|
27 |
model = T5ForConditionalGeneration.from_pretrained(model_name)
|
28 |
tokenizer = T5Tokenizer.from_pretrained(model_name)
|
|
|
29 |
# Example usage
|
30 |
input_text = " الاستاذ حزام جوبح يحصل على براعة اختراع في التعلم العميق"
|
31 |
output_text = model.generate(input_text)
|