vvv-knyazeva commited on
Commit
5bb9b8e
1 Parent(s): fe30f4d

Update pages/gpt_v1.py

Browse files
Files changed (1) hide show
  1. pages/gpt_v1.py +7 -7
pages/gpt_v1.py CHANGED
@@ -16,15 +16,15 @@ model = GPT2LMHeadModel.from_pretrained(
16
  model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
17
 
18
 
19
- length = st.sidebar.slider('Длина генерируемой последовательности:', 8, 256, 15)
20
- num_samples = st.sidebar.slider('Число генераций:', 1, 10, 1)
21
- temperature = st.sidebar.slider('Температура:', 1.0, 10.0, 2.0)
22
- top_k = st.sidebar.slider('Количество наиболее вероятных слов генерации:', 10, 200, 50)
23
- top_p = st.sidebar.slider('Минимальная суммарная вероятность топовых слов:', 0.4, 1.0, 0.9)
24
 
25
 
26
- prompt = st.text_input('Введите текст 👇:')
27
- if st.button('Сгенерировать текст'):
28
 
29
  with torch.inference_mode():
30
  prompt = tokenizer.encode(prompt, return_tensors='pt')
 
16
  model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
17
 
18
 
19
+ length = st.sidebar.slider('**Длина генерируемой последовательности:**', 8, 256, 15)
20
+ num_samples = st.sidebar.slider('**Число генераций:**', 1, 10, 1)
21
+ temperature = st.sidebar.slider('**Температура:**', 1.0, 10.0, 2.0)
22
+ top_k = st.sidebar.slider('**Количество наиболее вероятных слов генерации:**', 10, 200, 50)
23
+ top_p = st.sidebar.slider('**Минимальная суммарная вероятность топовых слов:**', 0.4, 1.0, 0.9)
24
 
25
 
26
+ prompt = st.text_input('**Введите текст 👇:**')
27
+ if st.button('**Сгенерировать текст**'):
28
 
29
  with torch.inference_mode():
30
  prompt = tokenizer.encode(prompt, return_tensors='pt')