Update pages/📖History_Mystery.py

pages/📖History_Mystery.py

@@ -59,8 +59,8 @@ st.title("""
  History Mystery
  """)
 # Добавление слайдера
-temp = st.slider("Градус дичи", 1.0, 20.0, 1.0)
-sen_quan = st.slider(" Длина сгенерированного отрывка", 2, 10, 5)
+temp = st.slider("Градус дичи", 1.0, 20.0, 5.0)
+sen_quan = st.slider(" Длина сгенерированного отрывка", 2, 10, 7)
 # Загрузка модели и токенизатора
 # model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
 # tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
@@ -87,8 +87,8 @@ 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=80, num_beams=5, do_sample=True,
-                            temperature=temp, top_k=50, top_p=0.6, no_repeat_ngram_size=4,
+    output = model_GPT.generate(input_ids=input_ids, max_length=100, num_beams=5, do_sample=True,
+                            temperature=temp, top_k=60, top_p=0.6, no_repeat_ngram_size=4,
                             num_return_sequences=sen_quan)
     # Декодирование сгенерированного текста
     generated_text = tokenizer_GPT.decode(output[0], skip_special_tokens=True)