Update summarizer.py

summarizer.py

@@ -4,16 +4,11 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, BartTokenizer, Ba
 eng_tokenizer = BartTokenizer.from_pretrained("facebook/bart-large-cnn")
 eng_model = BartForConditionalGeneration.from_pretrained("facebook/bart-large-cnn")
 
-# Русская модель
-rus_tokenizer = AutoTokenizer.from_pretrained("cointegrated/rut5-base-extsum")
-rus_model = AutoModelForSeq2SeqLM.from_pretrained("cointegrated/rut5-base-extsum")
-
+# Русская модель (публичная)
+rus_tokenizer = AutoTokenizer.from_pretrained("IlyaGusev/mbart_ru_sum_gazeta")
+rus_model = AutoModelForSeq2SeqLM.from_pretrained("IlyaGusev/mbart_ru_sum_gazeta")
 
 def generate_summary(text: str, lang="en", max_length=200, min_length=50) -> str:
-    """
-    lang="en" — английский текст
-    lang="ru" — русский текст
-    """
     if lang == "ru":
         tokenizer = rus_tokenizer
         model = rus_model