Update README.md

README.md

@@ -16,9 +16,7 @@ tags:
 - Text2Text Generation
 widget:
 - text: >-
-    خسارة مدوية لليفربول امام تولوز وفوز كبير لبيتيس، انتصار الفيولا واستون فيلا
-    في دوري المؤتمر، والد لويس دياز حر، فوز انديانا على ميلووكي, انتصار
-    للانترانيك
+    خسارة مدوية لليفربول امام تولوز وفوز كبير لبيتيس
 ---
 
 # ArabicT5: Classification and Generation of Arabic News
@@ -49,26 +47,27 @@ widget:
 
 # Example usage
 ```python
+
 from transformers import T5ForConditionalGeneration, T5Tokenizer, pipeline
-from arabert.preprocess import ArabertPreprocessor
 
-arabert_prep = ArabertPreprocessor(model_name="aubmindlab/bert-base-arabertv2")
 model_name="Hezam/arabic-T5-news-classification-generation"
 model = T5ForConditionalGeneration.from_pretrained(model_name)
 tokenizer = T5Tokenizer.from_pretrained(model_name)
 generation_pipeline = pipeline("text2text-generation",model=model,tokenizer=tokenizer)
 
-text = " خسارة مدوية لليفربول امام تولوز وفوز كبير لبيتيس، انتصار الفيولا واستون فيلا في دوري المؤتمر، والد لويس دياز حر، فوز انديانا على ميلووكي, انتصار للانترانيك"
+text = " خسارة مدوية لليفربول امام تولوز وفوز كبير لبيتيس"
 
-text_clean = arabert_prep.preprocess(text)
-g=generation_pipeline(text_clean,
+output= generation_pipeline(text,
                     num_beams=10,
-                    max_length=config.Generation_LEN,
+                    max_length=300,
                     top_p=0.9,
                     repetition_penalty = 3.0,
                     no_repeat_ngram_size = 3)[0]["generated_text"]
+
+print(output)
+4 كتبت لبنى عبد الله خسر فريق ليفربول بقيادة البرتغالي جوسفالدو فيريرا نظيره تولوز بنتيجة فى المباراة التى جمعتهما اليوم السبت ضمن منافسات الجولة الـ عمر مسابقة الدورى المصرى الممتاز والتى انتهت بفوز المارد الأحمر بهدف رد سجله أحمد سيد زيزو لاعب وسط الفريق الأول لكرة القدم بنادى
 ```
 
 ```bash
-output: 
+output: 4 كتبت لبنى عبد الله خسر فريق ليفربول بقيادة البرتغالي جوسفالدو فيريرا نظيره تولوز بنتيجة فى المباراة التى جمعتهما اليوم السبت ضمن منافسات الجولة الـ عمر مسابقة الدورى المصرى الممتاز والتى انتهت بفوز المارد الأحمر بهدف رد سجله أحمد سيد زيزو لاعب وسط الفريق الأول لكرة القدم بنادى
 ```