Update pipeline.py

pipeline.py

@@ -11,6 +11,7 @@ from sty import fg, bg, ef, rs, RgbBg, Style
 import re
 from tqdm import tqdm
 import gradio as gr
+import spaces
 
 dir_path = os.path.dirname(os.path.realpath(__file__)).split("\webapp")[0]
 os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
@@ -129,7 +130,7 @@ def get_pred_classification_labels(clauses, batch_size=32):
     return clause2labels
 
 
-
+@spaces.GPU
 def run_pipeline(text):
     snippets = auto_split(text)
     print(snippets)