Update app.py

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import spaces
 from transformers import pipeline
 
-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)
+token_skill_classifier = pipeline(model="jjzha/jobbert_skill_extraction", aggregation_strategy="first")
+token_knowledge_classifier = pipeline(model="jjzha/jobbert_knowledge_extraction", aggregation_strategy="first")
 
 
 examples = [
@@ -28,7 +28,6 @@ def aggregate_span(results):
 
     return new_results
 
-@spaces.GPU
 def ner(text):
     output_skills = token_skill_classifier(text)
     for result in output_skills: