Dzhamb commited on
Commit
498004d
1 Parent(s): a1ddf2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model.load_state_dict(torch.load('weight_model'))
19
 
20
  text = get_text(title, abstract)
21
  if text:
22
- raw_predictions = get_label(text)
23
  st.markdown(f"{raw_predictions}")
24
  else:
25
  st.markdown("Ваш запрос пуст. Введите хотя бы название")
 
19
 
20
  text = get_text(title, abstract)
21
  if text:
22
+ raw_predictions = get_labels(text, model, tokenizer)
23
  st.markdown(f"{raw_predictions}")
24
  else:
25
  st.markdown("Ваш запрос пуст. Введите хотя бы название")