Spaces:
Running
Running
Commit
·
cb2309a
1
Parent(s):
a2b85ee
Исправлен путь к модели
Browse files
app.py
CHANGED
@@ -60,34 +60,34 @@ def predict(title, summary, model, tokenizer, id2label, max_length=MAX_LEN, top_
|
|
60 |
return [(id2label[i], round(probs[i], 3)) for i in top_indices]
|
61 |
|
62 |
# Интерфейс Streamlit
|
63 |
-
st.title("
|
64 |
|
65 |
-
with st.expander("
|
66 |
st.markdown("""
|
67 |
-
Данная модель обучена на основе [SciBERT](https://huggingface.co/allenai/scibert_scivocab_uncased) для
|
68 |
|
69 |
-
- Использует **65 различных тегов** из тематик arXiv (например: `cs.CL`, `math.CO`, `stat.ML`, и т.д.), включая категорию other
|
70 |
- Модель обучена на **заголовках и аннотациях** научных публикаций
|
71 |
- На вход принимает **англоязычный текст**
|
72 |
- Предсказывает **топ-3 наиболее вероятных тега** для каждой статьи
|
73 |
|
74 |
-
Ниже вы можете посмотреть полный список возможных тегов
|
75 |
""")
|
76 |
|
77 |
with st.expander("📄 Список всех тегов"):
|
78 |
tag_list = sorted(label2id.keys())
|
79 |
st.markdown("\n".join([f"- `{tag}`" for tag in tag_list]))
|
80 |
|
81 |
-
st.write("Введите заголовок и аннотацию научной статьи (на английском)
|
82 |
|
83 |
title = st.text_input("**Title**")
|
84 |
summary = st.text_area("**Summary**", height=200)
|
85 |
|
86 |
-
if st.button("
|
87 |
if not title or not summary:
|
88 |
st.warning("Пожалуйста, введите и заголовок, и аннотацию!")
|
89 |
else:
|
90 |
preds = predict(title, summary, model, tokenizer, id2label)
|
91 |
-
st.subheader("
|
92 |
for tag, prob in preds:
|
93 |
st.write(f"**{tag}** — вероятность: `{prob:.3f}`")
|
|
|
60 |
return [(id2label[i], round(probs[i], 3)) for i in top_indices]
|
61 |
|
62 |
# Интерфейс Streamlit
|
63 |
+
st.title("ArXiv Tag Predictor")
|
64 |
|
65 |
+
with st.expander("Описание модели"):
|
66 |
st.markdown("""
|
67 |
+
Данная модель обучена на основе [SciBERT](https://huggingface.co/allenai/scibert_scivocab_uncased) для предсказаня первого тега статьей с сайта [arXiv.org](https://arxiv.org).
|
68 |
|
69 |
+
- Использует **65 различных тегов** из тематик arXiv (например: `cs.CL`, `math.CO`, `stat.ML`, и т.д.), включая категорию other, которая объединяет редкие теги.
|
70 |
- Модель обучена на **заголовках и аннотациях** научных публикаций
|
71 |
- На вход принимает **англоязычный текст**
|
72 |
- Предсказывает **топ-3 наиболее вероятных тега** для каждой статьи
|
73 |
|
74 |
+
Ниже вы можете посмотреть полный список возможных тегов
|
75 |
""")
|
76 |
|
77 |
with st.expander("📄 Список всех тегов"):
|
78 |
tag_list = sorted(label2id.keys())
|
79 |
st.markdown("\n".join([f"- `{tag}`" for tag in tag_list]))
|
80 |
|
81 |
+
st.write("Введите заголовок и аннотацию научной статьи (на английском):")
|
82 |
|
83 |
title = st.text_input("**Title**")
|
84 |
summary = st.text_area("**Summary**", height=200)
|
85 |
|
86 |
+
if st.button("Предсказать теги"):
|
87 |
if not title or not summary:
|
88 |
st.warning("Пожалуйста, введите и заголовок, и аннотацию!")
|
89 |
else:
|
90 |
preds = predict(title, summary, model, tokenizer, id2label)
|
91 |
+
st.subheader("Предсказанные теги:")
|
92 |
for tag, prob in preds:
|
93 |
st.write(f"**{tag}** — вероятность: `{prob:.3f}`")
|