Update pages/📖History_Mystery.py

pages/📖History_Mystery.py

@@ -9,7 +9,7 @@ st.title("""
  """)
 # Добавление слайдера
 temp = st.slider("Градус дичи", 1.0, 20.0, 1.0)
-sen_quan = st.slider(" Длина сгенерированного отрывка", 2, 30, 5)
+sen_quan = st.slider(" Длина сгенерированного отрывка", 2, 20, 5)
 # Загрузка модели и токенизатора
 # model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
@@ -36,7 +36,7 @@ def generate_text(model_GPT, tokenizer_GPT, prompt):
     # Преобразование входной строки в токены
     input_ids = tokenizer_GPT.encode(prompt, return_tensors='pt')
     # Генерация текста
-    output = model_GPT.generate(input_ids=input_ids, max_length=70, num_beams=5, do_sample=True,
+    output = model_GPT.generate(input_ids=input_ids, max_length=100, num_beams=5, do_sample=True,
                             temperature=temp, top_k=50, top_p=0.6, no_repeat_ngram_size=4,
                             num_return_sequences=sen_quan)
     # Декодирование сгенерированного текста
@@ -49,7 +49,7 @@ def main():
     # GPT-3 генерация текста
     """)
     # Ввод строки пользователем
-    prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века на руси")
+    prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века на Руси")
     # # Генерация текста по введенной строке
     # generated_text = generate_text(prompt)
     # Создание кнопки "Сгенерировать"