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()
|