Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,10 +18,10 @@ temperature = st.slider("Градус дичи", 1.0, 20.0, 1.0)
|
|
18 |
# tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
19 |
# #Задаем класс модели (уже в streamlit/tg_bot)
|
20 |
model = GPT2LMHeadModel.from_pretrained(
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
|
26 |
# # Вешаем сохраненные веса на нашу модель
|
27 |
model.load_state_dict(torch.load('model_hostory.pt'))
|
|
|
18 |
# tokenizer = GPT2Tokenizer.from_pretrained('sberbank-ai/rugpt3small_based_on_gpt2')
|
19 |
# #Задаем класс модели (уже в streamlit/tg_bot)
|
20 |
model = GPT2LMHeadModel.from_pretrained(
|
21 |
+
'sberbank-ai/rugpt3small_based_on_gpt2',
|
22 |
+
output_attentions = False,
|
23 |
+
output_hidden_states = False,
|
24 |
+
)
|
25 |
|
26 |
# # Вешаем сохраненные веса на нашу модель
|
27 |
model.load_state_dict(torch.load('model_hostory.pt'))
|