Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ for i in range(len(sorted_probs)):
|
|
92 |
sum_probs += sorted_probs[i]
|
93 |
if sum_probs > 0.95 or sorted_probs[i] < 0.001:
|
94 |
break
|
95 |
-
top_classes.append((
|
96 |
|
97 |
# Выводим список тем с их вероятностями
|
98 |
# from transformers import pipeline
|
|
|
92 |
sum_probs += sorted_probs[i]
|
93 |
if sum_probs > 0.95 or sorted_probs[i] < 0.001:
|
94 |
break
|
95 |
+
top_classes.append((tags_names[sorted_indices[i].item()], sorted_probs[i].item()))
|
96 |
|
97 |
# Выводим список тем с их вероятностями
|
98 |
# from transformers import pipeline
|