Hugging Face's logo

Spaces:
Linhz
/
ViMNer
Runtime error

App Files Community
Linhz commited on
Commit
3cfcf7e
•
1 Parent(s): 141ca4e

Update Model/NER/app_NER.py

Browse files
Files changed (1) hide show
  1. Model/NER/app_NER.py +2 -2
Model/NER/app_NER.py CHANGED
@@ -24,7 +24,7 @@ def show_ner():
24
  text = process_text(text)
25
  if st.button("Process NER"):
26
  if dataset == "VLSP2021":
27
- tagger = ViTagger(model_path='/Model/NER/VLSP2021/best_model.pt')
28
  a = text
29
  b = tagger.extract_entity_doc(a)
30
 
@@ -91,7 +91,7 @@ def show_ner():
91
  # print(html)
92
  st.markdown(html, unsafe_allow_html=True)
93
  elif dataset == "VLSP2016":
94
- tagger = ViTagger(model_path='/Model/NER/VLSP2016/best_model.pt')
95
  a = text
96
  b = tagger.extract_entity_doc(a)
97
 
 
24
  text = process_text(text)
25
  if st.button("Process NER"):
26
  if dataset == "VLSP2021":
27
+ tagger = ViTagger(model_path='Model/NER/VLSP2021/best_model.pt')
28
  a = text
29
  b = tagger.extract_entity_doc(a)
30
 
 
91
  # print(html)
92
  st.markdown(html, unsafe_allow_html=True)
93
  elif dataset == "VLSP2016":
94
+ tagger = ViTagger(model_path='Model/NER/VLSP2016/best_model.pt')
95
  a = text
96
  b = tagger.extract_entity_doc(a)
97