Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,9 @@
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
4 |
-
|
5 |
@st.cache_resource # кэширование
|
6 |
def load_model():
|
7 |
-
return pipeline("text-classification", model="voroninip/bert-paper-classifier-arxiv", top_k=None)
|
8 |
|
9 |
model = load_model()
|
10 |
|
@@ -18,7 +17,7 @@ def top_pct(preds, threshold=.95):
|
|
18 |
break
|
19 |
|
20 |
preds = preds[:(i+1)]
|
21 |
-
|
22 |
return preds
|
23 |
|
24 |
|
|
|
1 |
import streamlit as st
|
2 |
from transformers import pipeline
|
3 |
|
|
|
4 |
@st.cache_resource # кэширование
|
5 |
def load_model():
|
6 |
+
return pipeline("text-classification", model="voroninip/bert-paper-classifier-arxiv", top_k=None)
|
7 |
|
8 |
model = load_model()
|
9 |
|
|
|
17 |
break
|
18 |
|
19 |
preds = preds[:(i+1)]
|
20 |
+
|
21 |
return preds
|
22 |
|
23 |
|