Ilvir commited on
Commit
6dd30c8
·
1 Parent(s): 53a6520

Update gpt.py

Browse files
Files changed (1) hide show
  1. gpt.py +11 -6
gpt.py CHANGED
@@ -13,14 +13,19 @@ model = GPT2LMHeadModel.from_pretrained(
13
  model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
14
 
15
 
 
16
 
 
17
 
18
- prompt = st.text_input('Введите текст prompt:')
19
- length = st.slider('Длина генерируемой последовательности:', 8, 256, 15)
20
- num_samples = st.slider('Число генераций:', 1, 10, 1)
21
- temperature = st.slider('Температура:', 1.0, 10.0, 2.0)
22
- top_k = st.slider('Количество наиболее вероятных слов генерации:', 10, 200, 50)
23
- top_p = st.slider('Минимальная суммарная вероятность топовых слов:', 0.4, 1.0, 0.9)
 
 
 
24
 
25
 
26
 
 
13
  model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
14
 
15
 
16
+ col1, col2 = st.columns([7, 7])
17
 
18
+ with col2:
19
 
20
+ prompt = st.text_input('Введите текст prompt:')
21
+
22
+ with col1:
23
+
24
+ length = st.slider('Длина генерируемой последовательности:', 8, 256, 15)
25
+ num_samples = st.slider('Число генераций:', 1, 10, 1)
26
+ temperature = st.slider('Температура:', 1.0, 10.0, 2.0)
27
+ top_k = st.slider('Количество наиболее вероятных слов генерации:', 10, 200, 50)
28
+ top_p = st.slider('Минимальная суммарная вероятность топовых слов:', 0.4, 1.0, 0.9)
29
 
30
 
31