IkIzma commited on
Commit
d6d787f
·
1 Parent(s): 8b9c027

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -5,13 +5,11 @@ device = "cpu"
5
  tok, model = load_tokenizer_and_model("models/essays", device)
6
 
7
  st.markdown("## Генератор сочинений")
8
- st.markdown("Это приложение представляет из себя примитивный генератор сочинений. /\
9
- В качестве основной модели был взят трансформер RuGPT от Сбера https://huggingface.co/ai-forever/rugpt3small_based_on_gpt2. После этого на том же HF был скачан датасет с сочинениями на русском языке, который использовался для finetuning'а (см. ReadMe)")
10
-
11
- st.markdown("Для работы с приложением укажите максимальный размер сочинения, который хотите получить. Далее напишите тему. Для генерации сочинения длиной в 200 токенов потребуется подождать 6-7 минуты")
12
- st.markdown("Модель rugpt3small_based_on_gpt2 маленькая и датасет сочинений, на котором она добучалась, тоже маленький, поэтому не стоит расчитывать что получившееся сочинение будет хорошо отражать выбранную тему). /\
13
- Но, по крайней мере, текст получается осмысленными его стиль вполне соответствует стилю написания сочинений")
14
 
 
 
 
15
  with st.columns(3)[1]:
16
  st.markdown("<img width=200px src='https://ps-static.cdn-tinkoff.ru/static/ai-pushkin/portrait-2021-12-10-12-24-56.png'>", unsafe_allow_html=True)
17
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
 
5
  tok, model = load_tokenizer_and_model("models/essays", device)
6
 
7
  st.markdown("## Генератор сочинений")
8
+ st.markdown("Это приложение представляет из себя примитивный генератор сочинений. В качестве основной модели был взят трансформер RuGPT от Сбера https://huggingface.co/ai-forever/rugpt3small_based_on_gpt2. После этого на том же HF был скачан датасет с сочинениями на русском языке, который использовался для finetuning'а (см. ReadMe)")
 
 
 
 
 
9
 
10
+ st.markdown("Для работы с приложением укажите максимальный размер сочинения, который хотите получить. Далее напишите тему. Для генерации сочинения длиной в 200 токенов потребуется подождать 6-7 минут")
11
+ st.markdown("Модель rugpt3small_based_on_gpt2 маленькая и датасет сочинений, на котором она дообучалась, тоже маленький, поэтому не стоит рассчитывать, что получившееся сочинение будет хорошо отражать выбранную тему). Но, по крайней мере, текст получается осмысленным и его стиль вполне соответствует стилю написания сочинений.")
12
+ st.markdown("Чем меньше макисмальная длина, тем хуже получается итоговый результат. Относительно неплохие сочинения получаются при длине 500+, но по времни такая генерация занимает десятки минут")
13
  with st.columns(3)[1]:
14
  st.markdown("<img width=200px src='https://ps-static.cdn-tinkoff.ru/static/ai-pushkin/portrait-2021-12-10-12-24-56.png'>", unsafe_allow_html=True)
15
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter