Dzhamb commited on
Commit
415a8d6
1 Parent(s): 0126ae5

Update app.py

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