IkIzma commited on
Commit
9ba07dc
·
1 Parent(s): 39f9d41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,10 +9,11 @@ text = st.text_area("TEXT HERE")
9
  # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
10
  device = "cpu"
11
  thesis = "<s>Тема:" + "«" + text + "»." + "\nСочинение: "
 
12
 
13
  tok, model = load_tokenizer_and_model("models/essays", device)
14
  generated = generate(model, tok, thesis, device, num_beams=10)
15
 
16
  st.markdown(f"Hello!\n")
17
- st.markdown(generated[0])
18
  # выводим результаты модели в текстовое поле, на потеху пользователю
 
9
  # ^-- показать текстовое поле. В поле text лежит строка, которая находится там в данный момент
10
  device = "cpu"
11
  thesis = "<s>Тема:" + "«" + text + "»." + "\nСочинение: "
12
+ begin_index = len(thesis)
13
 
14
  tok, model = load_tokenizer_and_model("models/essays", device)
15
  generated = generate(model, tok, thesis, device, num_beams=10)
16
 
17
  st.markdown(f"Hello!\n")
18
+ st.markdown(generated[0][begin_index:])
19
  # выводим результаты модели в текстовое поле, на потеху пользователю