Update pages/📖History_Mystery.py

pages/📖History_Mystery.py

@@ -87,7 +87,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=100, num_beams=5, do_sample=True,
+    output = model_GPT.generate(input_ids=input_ids, max_length=100, num_beams=2*sen_quan, do_sample=True,
                             temperature=temp, top_k=60, top_p=0.6, no_repeat_ngram_size=4,
                             num_return_sequences=sen_quan)
     # Декодирование сгенерированного текста
@@ -100,7 +100,7 @@ def main():
     # GPT-3 генерация текста
     """)
     # Ввод строки пользователем
-    prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века на Руси")
+    prompt = st.text_area("Какую фразу нужно продолжить:", value="В средние века среди рыцарей")
     # # Генерация текста по введенной строке
     # generated_text = generate_text(prompt)
     # Создание кнопки "Сгенерировать"