Update README.md

README.md

@@ -33,6 +33,23 @@ For further details, see or [our demo site](https://juniper.nytud.hu/demo/nlp).
 | mT5 | 33.30/15.97/24.65 |
 | PEGASUS | 30.36/13.11/21.57 |
 
+## Usage
+
+```python
+from transformers import PegasusForConditionalGeneration, PegasusTokenizerFast
+model_name = 'NYTK/summarization-hi-pegasus-hungarian'
+
+tokenizer = PegasusTokenizerFast.from_pretrained(model_name)
+model = PegasusForConditionalGeneration.from_pretrained(model_name)
+input_text = "A Tisza-parti város állatkertjében régóta tartanak szurikátákat ( Suricata suricatta ) , de tavaly tavaszig nem sikerült szaporítani őket , annak ellenére , hogy tágas ház és kifutó épült számukra - közölte Veprik Róbert igazgató . 2010-ben alakult ki az új - három Amszterdamból származó nőstényből és egy budapesti fiatal hímből álló - csapat , amely szaporodni kezdett . 2011-ben három , idén pedig egy utóddal örvendeztették meg a gondozókat és az állatbarátokat . A szurikáták utódai - tizenegy hetes vemhesség után - október és március között vakon és szőrtelenül jönnek a világra . A kicsinyek háromhetesen bújnak elő az üregből , és nevelésükben mindkét szülő részt vesz . A szurikátacsapatokban a család tagjai nagyon szoros kapcsolatban állnak egymással , viszont nagyon harciasan fellépnek az idegenekkel szemben , akár meg is ölhetik azt az állatot , amelyet betolakodónak tekintenek . Bár a Dél-Afrikában , a Kalahári sivatagban őshonos cibetmacskaféle ragadozókat a szegedi állatkertben természetes élőhelyükhöz képest kevesebb veszély fenyegeti , a vadasparki erdőben ragadozó madarak is élnek , amelyek akár zsákmányként is tekinthetnének a szurikátákra . A szegedi csapatnál azonban szigorú őrség van , mindig lesi valaki két lábra állva a veszélyforrásokat ."
+
+tokenized_text = tokenizer(input_text, truncation=True, max_length=1024, return_tensors="pt")
+summarization = model.generate(**tokenized_text, max_length=256)
+
+print(tokenizer.batch_decode(summarization, skip_special_tokens=True))
+```
+
+
 ## Citation
 If you use this model, please cite the following paper: