1NEYRON1 commited on
Commit
e7531a2
·
1 Parent(s): c2eb36d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -167,11 +167,12 @@ def classify_text(title, description):
167
  st.error(f"Ошибка при классификации текста: {e}")
168
  return []
169
 
170
- readable_results = [
171
  (id_to_cat[int(entry['label'].split('_')[1])], entry['score'])
172
  for entry in results[0]
173
  ]
174
- return readable_results
 
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("Введите заголовок и краткое описание научной статьи, чтобы определить её тематические категории.")