Spaces:
Runtime error
Runtime error
Commit
·
a5e265e
1
Parent(s):
9499801
Update pages/gpt_v1.py
Browse files- pages/gpt_v1.py +2 -2
pages/gpt_v1.py
CHANGED
@@ -15,8 +15,8 @@ tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2
|
|
15 |
model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
|
16 |
|
17 |
prompt = st.text_input('Введите текст prompt:')
|
18 |
-
length = st.slider('Длина генерируемой последовательности:',
|
19 |
-
num_samples = st.slider('Число генераций:', 1,
|
20 |
temperature = st.slider('Температура:', 0.1, 1.0, 0.5)
|
21 |
|
22 |
|
|
|
15 |
model.load_state_dict(torch.load('model.pt', map_location=torch.device('cpu')))
|
16 |
|
17 |
prompt = st.text_input('Введите текст prompt:')
|
18 |
+
length = st.slider('Длина генерируемой последовательности:', 1, 256, 16)
|
19 |
+
num_samples = st.slider('Число генераций:', 1, 6, 1)
|
20 |
temperature = st.slider('Температура:', 0.1, 1.0, 0.5)
|
21 |
|
22 |
|