Zhdantim commited on
Commit
4707d05
·
verified ·
1 Parent(s): bba0e26

some debug

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -20,9 +20,9 @@ def load_tokenizer():
20
  return AutoTokenizer.from_pretrained('microsoft/deberta-v3-small')
21
 
22
 
23
- model = load_model()
24
- id2classes = load_id2classes()
25
- tokenizer = load_tokenizer()
26
 
27
 
28
  def get_top_classes(text):
@@ -48,9 +48,10 @@ if st.button('Submit'):
48
  else:
49
  text = title.title()
50
 
51
- top_classes, probs = get_top_classes(text)
52
 
53
- for p, cls in zip(probs, top_classes):
54
- st.success(f'Статья относится к {cls} с вероятностью {p}')
 
55
  else:
56
  st.error('Введите название статьи')
 
20
  return AutoTokenizer.from_pretrained('microsoft/deberta-v3-small')
21
 
22
 
23
+ #model = load_model()
24
+ #id2classes = load_id2classes()
25
+ #tokenizer = load_tokenizer()
26
 
27
 
28
  def get_top_classes(text):
 
48
  else:
49
  text = title.title()
50
 
51
+ #top_classes, probs = get_top_classes(text)
52
 
53
+ #for p, cls in zip(probs, top_classes):
54
+ #st.success(f'Статья относится к {cls} с вероятностью {p}')
55
+ st.success('Запустилось')
56
  else:
57
  st.error('Введите название статьи')