Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 import spaces
 from transformers import pipeline
 
-token_skill_classifier = pipeline(model="jjzha/jobbert_skill_extraction", aggregation_strategy="first")
-token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="first")
-token_skill_classifier.to("cuda")
-token_knowledge_classifier.to("cuda")
+token_skill_classifier = pipeline(model="jjzha/jobbert_skill_extraction", aggregation_strategy="first", device=0)
+token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="first", device=0)
+
 
 examples = [
         "Knowing Python is a plus",