Mike commited on
Commit
21d9527
1 Parent(s): 1da09fe

add aggregation step

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- token_skill_classifier = pipeline(model="jjzha/jobbert_skill_extraction")
5
- token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction")
6
 
7
 
8
  examples = [
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ token_skill_classifier = pipeline(model="jjzha/jobbert_skill_extraction", aggregation_strategy="simple")
5
+ token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="simple")
6
 
7
 
8
  examples = [