upd

pages/2_👨‍🏫_Начало_работы.py

@@ -144,7 +144,7 @@ if st.session_state.get('-LOGGED_IN_BOOL-'):
             key='-SAVE_IN_CLOUD-')
         DOWNLOAD_VERSION = col2.checkbox(
             label='**Скачать**',
-            value=False,
+            value=True,
             key='-DOWNLOAD_VERSION-')
         ONLINE_TEST_VERSION = col3.checkbox(
             label='**Онлайн тест**',

utilities_language_bert/rus_main_workflow_bert.py

@@ -232,7 +232,7 @@ def main_workflow(
                 f'\n\n{"$" * 70}\n\n{PREPARED_TASKS["KEYS_ONLY"]}'
     logs.update(label='Сейчас все будет готово!', state='running')
     progress.progress(95)
-    save_name = save_name if save_name != '' else f'{str(datetime.datetime.now())[:-7]}_{original_text[:20]}'
+    save_name = save_name if save_name != '' else f'{str(datetime.datetime.now())[:-7]}_{original_text[:20]}_{level}_M3'
     out = {
         'name': save_name,
         'STUDENT_OUT': STUDENT_OUT,

utilities_language_general/rus_constants.py

@@ -6,6 +6,8 @@ import streamlit as st
 from pickle import load
 from transformers import pipeline
 from summarizer import Summarizer
+from torch import cuda, device
+device = device('cuda' if cuda.is_available else 'cpu')
 
 
 @st.cache_resource
@@ -34,7 +36,7 @@ def load_spacy():
 @st.cache_resource
 def load_bert():
     with st.spinner('Загружаю языковую модель'):
-        _pipeline = pipeline(task="fill-mask", model="a-v-bely/ruBert-base-finetuned-russian-moshkov-child-corpus-pro")
+        _pipeline = pipeline(task="fill-mask", model="a-v-bely/ruBert-base-finetuned-russian-moshkov-child-corpus-pro", device=device)
     return _pipeline
 
 

utilities_language_w2v/rus_main_workflow_w2v.py

@@ -242,7 +242,8 @@ def main_workflow(
                 f'\n\n{"$" * 70}\n\n{PREPARED_TASKS["KEYS_ONLY"]}'
     logs.update(label='Сейчас все будет готово!', state='running')
     progress.progress(90)
-    save_name = save_name if save_name != '' else f'{str(datetime.datetime.now())[:-7]}_{original_text[:20]}'
+    md = {'Модель-1': 'M1', 'Модель-2': 'M2'}
+    save_name = save_name if save_name != '' else f'{str(datetime.datetime.now())[:-7]}_{original_text[:20]}_{level}_{md[model_name]}'
     out = {
         'name': save_name,
         'STUDENT_OUT': STUDENT_OUT,