SaviAnna commited on
Commit
ce7fad9
·
1 Parent(s): 44cac7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- # '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'))
 
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'))