jjzha commited on
Commit
9e7b09f
1 Parent(s): 4413a8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -2,10 +2,8 @@ import gradio as gr
2
  import spaces
3
  from transformers import pipeline
4
 
5
- token_skill_classifier = pipeline(model="jjzha/escoxlmr_skill_extraction", aggregation_strategy="first")
6
- token_knowledge_classifier = pipeline(model="jjzha/escoxlmr_knowledge_extraction", aggregation_strategy="first")
7
- token_skill_classifier.to("cuda")
8
- token_knowledge_classifier.to("cuda")
9
 
10
 
11
  examples = [
 
2
  import spaces
3
  from transformers import pipeline
4
 
5
+ token_skill_classifier = pipeline(model="jjzha/escoxlmr_skill_extraction", aggregation_strategy="first", device=0)
6
+ token_knowledge_classifier = pipeline(model="jjzha/escoxlmr_knowledge_extraction", aggregation_strategy="first", device=0)
 
 
7
 
8
 
9
  examples = [