Update app.py
Browse files
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()
|