Update app.py
Browse files
app.py
CHANGED
|
@@ -141,6 +141,7 @@ id_to_cat = {0: 'High Energy Physics - Theory',
|
|
| 141 |
137: 'Disordered Systems and Neural Networks'
|
| 142 |
}
|
| 143 |
|
|
|
|
| 144 |
def load_model():
|
| 145 |
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
|
| 146 |
model = AutoModelForSequenceClassification.from_pretrained(
|
|
@@ -173,7 +174,7 @@ def classify_text(title, description):
|
|
| 173 |
return [(label, score / total) for label, score in res]
|
| 174 |
|
| 175 |
st.title("🔬 Классификация научных статей")
|
| 176 |
-
st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")
|
| 177 |
|
| 178 |
title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
|
| 179 |
description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")
|
|
|
|
| 141 |
137: 'Disordered Systems and Neural Networks'
|
| 142 |
}
|
| 143 |
|
| 144 |
+
@st.cache_resource
|
| 145 |
def load_model():
|
| 146 |
tokenizer = AutoTokenizer.from_pretrained('distilbert-base-cased')
|
| 147 |
model = AutoModelForSequenceClassification.from_pretrained(
|
|
|
|
| 174 |
return [(label, score / total) for label, score in res]
|
| 175 |
|
| 176 |
st.title("🔬 Классификация научных статей")
|
| 177 |
+
st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории. (сервис работает только для англоязычных статей)")
|
| 178 |
|
| 179 |
title = st.text_input("📝 Заголовок статьи", placeholder="Например: Deep Learning for Image Recognition")
|
| 180 |
description = st.text_area("🧾 Краткое описание статьи", height=150, placeholder="Кратко опишите содержание статьи...")
|