stal76 commited on
Commit
b8dd604
·
1 Parent(s): 38083c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -53,8 +53,12 @@ def BuildAnswer(txt):
53
  8: "Statistics"
54
  }
55
 
56
- model, tokenizer, model_emb = GetModelAndTokenizer()
 
 
57
 
 
 
58
  embed = get_word_vector(txt, tokenizer, model_emb)
59
  logits = torch.nn.functional.softmax(model(embed), dim=0)
60
  best_tags = torch.argsort(logits, descending=True)
@@ -77,7 +81,7 @@ st.markdown("<img width=200px src='https://rozetked.me/images/uploads/dwoilp3BVj
77
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
78
 
79
  title = st.text_area("Title:")
80
- abstract = st.text_area("Abstract:", height=10)
81
 
82
  #from transformers import pipeline
83
  #pipe = pipeline("ner", "Davlan/distilbert-base-multilingual-cased-ner-hrl")
 
53
  8: "Statistics"
54
  }
55
 
56
+ txt = txt.strip()
57
+ if txt == '':
58
+ return ''
59
 
60
+ model, tokenizer, model_emb = GetModelAndTokenizer()
61
+
62
  embed = get_word_vector(txt, tokenizer, model_emb)
63
  logits = torch.nn.functional.softmax(model(embed), dim=0)
64
  best_tags = torch.argsort(logits, descending=True)
 
81
  # ^-- можно показывать пользователю текст, картинки, ограниченное подмножество html - всё как в jupyter
82
 
83
  title = st.text_area("Title:")
84
+ abstract = st.text_area("Abstract:", height=100)
85
 
86
  #from transformers import pipeline
87
  #pipe = pipeline("ner", "Davlan/distilbert-base-multilingual-cased-ner-hrl")