SaviAnna commited on
Commit
3d83edc
·
1 Parent(s): c447610

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 Mistery
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.0)
 
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.0, top_k=50, top_p=0.6, no_repeat_ngram_size=3,
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
  # Декодирование сгенерированного текста