ElijahDi commited on
Commit
ae2bd9e
·
verified ·
1 Parent(s): 0a3c50b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -190,7 +190,6 @@ elif selected_model == "Генерация текста GPT-моделью по
190
  tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
191
  model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
192
  model.load_state_dict(torch.load('model_dict.pt', map_location=device))
193
- model.eval()
194
 
195
  if st.button('Сделать гороскоп'):
196
  start_time = time.time()
 
190
  tokenizer = GPT2Tokenizer.from_pretrained(model_name_or_path)
191
  model = GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
192
  model.load_state_dict(torch.load('model_dict.pt', map_location=device))
 
193
 
194
  if st.button('Сделать гороскоп'):
195
  start_time = time.time()