Dagobert42 commited on
Commit
a5515f4
·
1 Parent(s): 4330a70

use better model for NER

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. helpers.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ sentence = choice(sentences)
11
  st.title("Semantic Frame Augmentation")
12
  st.subheader("Analysing difficult low-resource domains with only a handful of examples")
13
 
14
- st.write("This space uses a google/mobilebert-uncased model for NER")
15
  augment = st.toggle('Use augmented model for NER', value=False)
16
 
17
  txt = st.text_area(
 
11
  st.title("Semantic Frame Augmentation")
12
  st.subheader("Analysing difficult low-resource domains with only a handful of examples")
13
 
14
+ st.write("This space uses a xlnet-base-cased model for NER")
15
  augment = st.toggle('Use augmented model for NER', value=False)
16
 
17
  txt = st.text_area(
helpers.py CHANGED
@@ -1,11 +1,11 @@
1
  from transformers import pipeline
2
 
3
  baseline_classifier = pipeline("ner",
4
- model="Dagobert42/biored-finetuned",
5
  aggregation_strategy="simple"
6
  )
7
  augmented_classifier = pipeline("ner",
8
- model="Dagobert42/biored-augmented",
9
  aggregation_strategy="simple"
10
  )
11
 
 
1
  from transformers import pipeline
2
 
3
  baseline_classifier = pipeline("ner",
4
+ model="Dagobert42/xlnet-base-cased-biored-augmented",
5
  aggregation_strategy="simple"
6
  )
7
  augmented_classifier = pipeline("ner",
8
+ model="Dagobert42/xlnet-base-cased-biored-finetuned",
9
  aggregation_strategy="simple"
10
  )
11