ElijahDi commited on
Commit
6e92274
·
verified ·
1 Parent(s): 0838f0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -17
app.py CHANGED
@@ -191,20 +191,23 @@ elif selected_model == "Генерация текста GPT-моделью по
191
  tokenizer = GPT2Tokenizer.from_pretrained(path)
192
 
193
  if st.button('Сделать гороскоп'):
194
- start_time = time.time()
195
- input_ids = tokenizer.encode(user_text_input, return_tensors="pt").to(device)
196
- model.eval()
197
- with torch.no_grad():
198
- out = model.generate(input_ids,
199
- do_sample=True,
200
- num_beams=2,
201
- temperature=1.1,
202
- top_p=0.9,
203
- max_length=50,
204
- )
205
-
206
- generated_text = list(map(tokenizer.decode, out))[0]
207
- end_time = time.time()
208
- prediction_time = end_time - start_time
209
-
210
- st.write(f'Ваше предсказание: {generated_text}')
 
 
 
 
191
  tokenizer = GPT2Tokenizer.from_pretrained(path)
192
 
193
  if st.button('Сделать гороскоп'):
194
+ with st.spinner('Генерация текста...'):
195
+ start_time = time.time()
196
+ input_ids = tokenizer.encode(user_text_input, return_tensors="pt").to(device)
197
+ model.eval()
198
+ with torch.no_grad():
199
+ out = model.generate(
200
+ input_ids,
201
+ do_sample=True,
202
+ num_beams=2,
203
+ temperature=1.1,
204
+ top_p=0.9,
205
+ max_length=50,
206
+ )
207
+
208
+ generated_text = tokenizer.decode(out[0], skip_special_tokens=True)
209
+ end_time = time.time()
210
+ prediction_time = end_time - start_time
211
+
212
+ st.success('Готово!')
213
+ st.write({generated_text})