Spaces:
Sleeping
Sleeping
Rename pages/✨third.py to pages/📖History_Mystery.py
Browse files
pages/{✨third.py → 📖History_Mystery.py}
RENAMED
@@ -6,13 +6,14 @@ from PIL import Image
|
|
6 |
import torch
|
7 |
|
8 |
st.title("""
|
9 |
-
History
|
10 |
""")
|
11 |
# image = Image.open('data-scins.jpeg')
|
12 |
|
13 |
# st.image(image, caption='Current mood')
|
14 |
# Добавление слайдера
|
15 |
-
temperature = st.slider("Градус дичи", 1.0, 20.0, 1
|
|
|
16 |
# Загрузка модели и токенизатора
|
17 |
# model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
18 |
# tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
@@ -37,7 +38,7 @@ def generate_text(prompt):
|
|
37 |
|
38 |
# Генерация текста
|
39 |
output = model.generate(input_ids=input_ids, max_length=70, num_beams=5, do_sample=True,
|
40 |
-
temperature=1
|
41 |
num_return_sequences=3)
|
42 |
|
43 |
# Декодирование сгенерированного текста
|
|
|
6 |
import torch
|
7 |
|
8 |
st.title("""
|
9 |
+
History Mystery
|
10 |
""")
|
11 |
# image = Image.open('data-scins.jpeg')
|
12 |
|
13 |
# st.image(image, caption='Current mood')
|
14 |
# Добавление слайдера
|
15 |
+
temperature = st.slider("Градус дичи", 1.0, 20.0, 1)
|
16 |
+
max_len = st.slider(" Длина сгенерированного отрывка", 40, 120, 70)
|
17 |
# Загрузка модели и токенизатора
|
18 |
# model = GPT2LMHeadModel.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
19 |
# tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
|
|
38 |
|
39 |
# Генерация текста
|
40 |
output = model.generate(input_ids=input_ids, max_length=70, num_beams=5, do_sample=True,
|
41 |
+
temperature=1, top_k=50, top_p=0.6, no_repeat_ngram_size=3,
|
42 |
num_return_sequences=3)
|
43 |
|
44 |
# Декодирование сгенерированного текста
|