Update pages/gpt.py

pages/gpt.py

@@ -18,14 +18,8 @@ model.load_state_dict(torch.load('models/modelgpt.pt', map_location=torch.device
 
 
 length = st.sidebar.slider('**Длина генерируемой последовательности:**', 8, 256, 15)
-
-num_samples = st.selectbox(
-    '**Число генераций:**',
-    ('1', '2', '3', '4', '5'))
-
-
-# num_samples = st.sidebar.slider('**Число генераций:**', 1, 10, 1)
-temperature = st.sidebar.slider('**Температура:**', 1.0, 10.0, 2.0)
+num_samples = st.sidebar.slider('**Число генераций:**', 1, 10, 1)
+temperature = st.sidebar.slider('**Температура:**', 1, 10, 3)
 top_k = st.sidebar.slider('**Количество наиболее вероятных слов генерации:**', 10, 200, 50)
 top_p = st.sidebar.slider('**Минимальная суммарная вероятность топовых слов:**', 0.4, 1.0, 0.9)
 
@@ -44,7 +38,7 @@ if st.button('**Сгенерировать текст**'):
             top_k=top_k,
             top_p=top_p,
             no_repeat_ngram_size=3,
-            num_return_sequences=int(num_samples),
+            num_return_sequences=num_samples,
             ).cpu().numpy()
         st.write('**_Результат_** 👇')
         for i, out_ in enumerate(out):