Ilvir commited on
Commit
43ed82a
·
1 Parent(s): e4b5631

Update gpt.py

Browse files
Files changed (1) hide show
  1. gpt.py +4 -1
gpt.py CHANGED
@@ -13,7 +13,7 @@ model = GPT2LMHeadModel.from_pretrained(
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 col1:
19
 
@@ -24,6 +24,9 @@ with col1:
24
  top_p = st.slider('Минимальная суммарная вероятность топовых слов:', 0.4, 1.0, 0.9)
25
 
26
  with col2:
 
 
 
27
 
28
  prompt = st.text_input('Введите текст prompt:')
29
 
 
13
  model.load_state_dict(torch.load('modelgpt.pt', map_location=torch.device('cpu')))
14
 
15
 
16
+ col1, col2, col3 = st.columns([7, 1, 7])
17
 
18
  with col1:
19
 
 
24
  top_p = st.slider('Минимальная суммарная вероятность топовых слов:', 0.4, 1.0, 0.9)
25
 
26
  with col2:
27
+ pass
28
+
29
+ with col3:
30
 
31
  prompt = st.text_input('Введите текст prompt:')
32