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="Кратко опишите содержание статьи...")
|