sberbank-ai/ruT5-baseaglazkova/keyphrase_extraction_russian1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2tok = AutoTokenizer.from_pretrained("denis-gordeev/rut5-keyphrase-ru", use_fast=False)
3mdl = AutoModelForSeq2SeqLM.from_pretrained("denis-gordeev/rut5-keyphrase-ru")
4
5text = "keyphrase: Новая модель обнаруживает аномалии в банковских транзакциях..."
6inp = tok(text, return_tensors="pt")
7out = mdl.generate(**inp, max_new_tokens=64, num_beams=4, no_repeat_ngram_size=3)
8print(tok.decode(out[0], skip_special_tokens=True))