SaviAnna commited on
Commit
5a41c9a
·
1 Parent(s): ce7fad9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ model = GPT2LMHeadModel.from_pretrained(
24
  )
25
 
26
  # # Вешаем сохраненные веса на нашу модель
27
- model.load_state_dict(torch.load('model_hostory.pt'))
28
  # Функция для генерации текста
29
  def generate_text(prompt):
30
  # Преобразование входной строки в токены
 
24
  )
25
 
26
  # # Вешаем сохраненные веса на нашу модель
27
+ model.load_state_dict(torch.load('model_history.pt'))
28
  # Функция для генерации текста
29
  def generate_text(prompt):
30
  # Преобразование входной строки в токены