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 |
|