Update app.py

app.py

@@ -13,7 +13,7 @@ summarization_model_name = 'sarahai/ruT5-base-summarizer'
 summarization_model = T5ForConditionalGeneration.from_pretrained(summarization_model_name)
 summarization_tokenizer = T5Tokenizer.from_pretrained(summarization_model_name)
 
-def translate(text, translation_model, translation_tokenizer, src_lang='uzb_Cyrl', tgt_lang='rus_Cyrl', a=16, b=1.5, max_input_length=1024):
+def translate(text, translation_model, translation_tokenizer, src_lang='uzb_Cyrl', tgt_lang='rus_Cyrl', a=16, b=1.5, max_input_length=2048):
     translation_tokenizer.src_lang = src_lang
     translation_tokenizer.tgt_lang = tgt_lang
     inputs = translation_tokenizer(text, return_tensors='pt', padding=True, truncation=True, max_length=max_input_length)
@@ -25,8 +25,8 @@ def translate(text, translation_model, translation_tokenizer, src_lang='uzb_Cyrl
     translated_text = translation_tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_text
 
-def summarize(translated_text, summarization_model, summarization_tokenizer, max_length=250, min_length=150):
-    input_ids = summarization_tokenizer.encode("summarize: " + translated_text, return_tensors="pt", max_length=1024, truncation=True)
+def summarize(translated_text, summarization_model, summarization_tokenizer, max_length=250):
+    input_ids = summarization_tokenizer.encode("summarize: " + translated_text, return_tensors="pt", max_length=2048, truncation=True)
     summary_ids = summarization_model.generate(
         input_ids,
         max_length=max_length,
@@ -49,7 +49,7 @@ if st.button("Перевести и суммаризировать"):
             st.text_area("Переведенный текст (на русском):", value=translated_text, height=200)
         
         with st.spinner('Суммаризируем...'):
-            summary_text = summarize(translated_text, summarization_model, summarization_tokenizer, max_length=250, min_length=150)
+            summary_text = summarize(translated_text, summarization_model, summarization_tokenizer, max_length=250)
             st.text_area("Суммаризация (на русском):", value=summary_text, height=100)
     else:
         st.warning("Пожалуйста, введите текст на узбекском языке для перевода.")