Update app.py

app.py

@@ -8,7 +8,6 @@ import re
 import os
 from transformers import pipeline
 
-@spaces.GPU
 pipe = pipeline("text-generation", model="jslin09/gemma2-2b-ner", device="cuda")
 
 elements = {'LEO_SOC': ('犯罪主體', 'Subject of Crime'),
@@ -134,6 +133,7 @@ def tag_in_color(tag_content, tag):
 #    print(parts)
     return parts
 
+@spaces.GPU
 def le_tagger(content, tag):
     jaccard_score = [] # 紀錄單篇的分數
     score = ()