Update app.py
Browse files
app.py
CHANGED
@@ -53,8 +53,12 @@ def BuildAnswer(txt):
|
|
53 |
8: "Statistics"
|
54 |
}
|
55 |
|
56 |
-
|
|
|
|
|
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=
|
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")
|