Update pages/gpt.py
Browse files- pages/gpt.py +6 -0
pages/gpt.py
CHANGED
|
@@ -18,8 +18,14 @@ model.load_state_dict(torch.load('models/modelgpt.pt', map_location=torch.device
|
|
| 18 |
|
| 19 |
|
| 20 |
length = st.sidebar.slider('**Длина генерируемой последовательности:**', 8, 256, 15)
|
|
|
|
|
|
|
| 21 |
num_samples = st.sidebar.slider('**Число генераций:**', 1, 10, 1)
|
|
|
|
|
|
|
| 22 |
temperature = st.sidebar.slider('**Температура:**', 0.1, 10.0, 3.0)
|
|
|
|
|
|
|
| 23 |
top_k = st.sidebar.slider('**Количество наиболее вероятных слов генерации:**', 10, 200, 50)
|
| 24 |
top_p = st.sidebar.slider('**Минимальная суммарная вероятность топовых слов:**', 0.4, 1.0, 0.9)
|
| 25 |
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
length = st.sidebar.slider('**Длина генерируемой последовательности:**', 8, 256, 15)
|
| 21 |
+
if length > 100:
|
| 22 |
+
st.warning("this is very hard for me, please have pity on me", icon="🤖")
|
| 23 |
num_samples = st.sidebar.slider('**Число генераций:**', 1, 10, 1)
|
| 24 |
+
if num_samples > 4:
|
| 25 |
+
st.warning("OH MY ..., I have to work late again!!!", icon="🤖")
|
| 26 |
temperature = st.sidebar.slider('**Температура:**', 0.1, 10.0, 3.0)
|
| 27 |
+
if num_samples > 6.0:
|
| 28 |
+
st.info('What? You want to get some kind of bullshit as a result?', icon="🤖")
|
| 29 |
top_k = st.sidebar.slider('**Количество наиболее вероятных слов генерации:**', 10, 200, 50)
|
| 30 |
top_p = st.sidebar.slider('**Минимальная суммарная вероятность топовых слов:**', 0.4, 1.0, 0.9)
|
| 31 |
|