Update app.py
Browse files
app.py
CHANGED
@@ -186,7 +186,7 @@ elif selected_model == "Генерация текста GPT-моделью по
|
|
186 |
user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
|
187 |
|
188 |
# GPT2
|
189 |
-
path = '
|
190 |
model = GPT2LMHeadModel.from_pretrained(path).to(device)
|
191 |
tokenizer = GPT2Tokenizer.from_pretrained(path)
|
192 |
|
|
|
186 |
user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
|
187 |
|
188 |
# GPT2
|
189 |
+
path = 'model_pretra/'
|
190 |
model = GPT2LMHeadModel.from_pretrained(path).to(device)
|
191 |
tokenizer = GPT2Tokenizer.from_pretrained(path)
|
192 |
|