IkIzma commited on
Commit
e50a355
·
1 Parent(s): b29330a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,8 +4,9 @@ from Generate_text import generate, load_tokenizer_and_model
4
  device = "cpu"
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")
 
9
  with st.columns(3)[1]:
10
  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)
11
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
 
4
  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
  with st.columns(3)[1]:
11
  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)
12
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter