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