ElijahDi commited on
Commit
9d2b8d4
·
verified ·
1 Parent(s): d8f585b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -186,8 +186,9 @@ elif selected_model == "Генерация текста GPT-моделью по
186
  user_text_input = st.text_area('Введите информацию о себе для формиорования гороскопа:')
187
 
188
  # GPT2
189
- model = GPT2LMHeadModel.from_pretrained("model_pretra.zip").to(device)
190
- tokenizer = GPT2Tokenizer.from_pretrained("tokenizer.pt.zip")
 
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()