Dzhamb commited on
Commit
4ccb522
1 Parent(s): b53e809

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import streamlit as st
2
 
3
  st.markdown("## Классификатор статей")
4
- st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVjlE.jpg'>", unsafe_allow_html=True)
5
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
6
 
7
  title = st.text_area("Введите название статьи")
@@ -10,7 +10,7 @@ abstract = st.text_area("Введите аннотацию к статье, abst
10
 
11
  from transformers import pipeline
12
  pipe = pipeline("ner", "Davlan/distilbert-base-multilingual-cased-ner-hrl")
13
- raw_predictions = pipe(text)
14
  # тут уже знакомый вам код с huggingface.transformers -- его можно заменить на что угодно от fairseq до catboost
15
 
16
  st.markdown(f"{raw_predictions}")
 
1
  import streamlit as st
2
 
3
  st.markdown("## Классификатор статей")
4
+ st.markdown("Сервис классифицирует статьи по названию и аннотации. Нужно ввести в каждое окошко свою сущность и вам выдадут к какому классу относится статья")
5
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
6
 
7
  title = st.text_area("Введите название статьи")
 
10
 
11
  from transformers import pipeline
12
  pipe = pipeline("ner", "Davlan/distilbert-base-multilingual-cased-ner-hrl")
13
+ raw_predictions = pipe(title + ' ' + abstract)
14
  # тут уже знакомый вам код с huggingface.transformers -- его можно заменить на что угодно от fairseq до catboost
15
 
16
  st.markdown(f"{raw_predictions}")