Update app.py

app.py

@@ -2,10 +2,8 @@ import gradio as gr
 import spaces
 from transformers import pipeline
 
-token_skill_classifier = pipeline(model="jjzha/escoxlmr_skill_extraction", aggregation_strategy="first")
-token_knowledge_classifier = pipeline(model="jjzha/escoxlmr_knowledge_extraction", aggregation_strategy="first")
-token_skill_classifier.to("cuda")
-token_knowledge_classifier.to("cuda")
+token_skill_classifier = pipeline(model="jjzha/escoxlmr_skill_extraction", aggregation_strategy="first", device=0)
+token_knowledge_classifier = pipeline(model="jjzha/escoxlmr_knowledge_extraction", aggregation_strategy="first", device=0)
 
 
 examples = [