Hugging Face's logo

Spaces:
jjzha
/
skill_extraction_demo
Running

App Files Community
jjzha commited on
Commit
3c343d3
·
verified ·
1 Parent(s): e72157c

Update app.py

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", aggregation_strategy="first")
5
- token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="first")
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="first", device="cuda")
5
+ token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="first", device="cuda")
6
 
7
 
8
  examples = [