Update app.py
Browse files
app.py
CHANGED
|
@@ -186,8 +186,9 @@ elif selected_model == "Генерация текста GPT-моделью по
|
|
| 186 |
user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
|
| 187 |
|
| 188 |
# GPT2
|
| 189 |
-
|
| 190 |
-
|
|
|
|
| 191 |
|
| 192 |
if st.button('Сделать гороскоп'):
|
| 193 |
start_time = time.time()
|
|
|
|
| 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 |
|
| 193 |
if st.button('Сделать гороскоп'):
|
| 194 |
start_time = time.time()
|