MikhailPugachev commited on
Commit
cb2309a
·
1 Parent(s): a2b85ee

Исправлен путь к модели

Browse files
Files changed (1) hide show
  1. app.py +8 -8
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("🧠 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}`")
 
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}`")