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