voroninip commited on
Commit
53ac886
·
verified ·
1 Parent(s): 708e41e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
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