Update app.py
Browse files
app.py
CHANGED
@@ -167,11 +167,12 @@ def classify_text(title, description):
|
|
167 |
st.error(f"Ошибка при классификации текста: {e}")
|
168 |
return []
|
169 |
|
170 |
-
|
171 |
(id_to_cat[int(entry['label'].split('_')[1])], entry['score'])
|
172 |
for entry in results[0]
|
173 |
]
|
174 |
-
|
|
|
175 |
|
176 |
st.title("🔬 Классификация научных статей")
|
177 |
st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")
|
|
|
167 |
st.error(f"Ошибка при классификации текста: {e}")
|
168 |
return []
|
169 |
|
170 |
+
res = [
|
171 |
(id_to_cat[int(entry['label'].split('_')[1])], entry['score'])
|
172 |
for entry in results[0]
|
173 |
]
|
174 |
+
total = sum(score for _, score in res)
|
175 |
+
return [(label, score / total) for label, score in res]
|
176 |
|
177 |
st.title("🔬 Классификация научных статей")
|
178 |
st.markdown("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")
|