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("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")
|