Update pipeline.py
Browse files- pipeline.py +0 -2
pipeline.py
CHANGED
|
@@ -11,7 +11,6 @@ from sty import fg, bg, ef, rs, RgbBg, Style
|
|
| 11 |
import re
|
| 12 |
from tqdm import tqdm
|
| 13 |
import gradio as gr
|
| 14 |
-
import spaces
|
| 15 |
|
| 16 |
dir_path = os.path.dirname(os.path.realpath(__file__)).split("\webapp")[0]
|
| 17 |
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
|
@@ -130,7 +129,6 @@ def get_pred_classification_labels(clauses, batch_size=32):
|
|
| 130 |
return clause2labels
|
| 131 |
|
| 132 |
|
| 133 |
-
@spaces.GPU
|
| 134 |
def run_pipeline(text):
|
| 135 |
snippets = auto_split(text)
|
| 136 |
print(snippets)
|
|
|
|
| 11 |
import re
|
| 12 |
from tqdm import tqdm
|
| 13 |
import gradio as gr
|
|
|
|
| 14 |
|
| 15 |
dir_path = os.path.dirname(os.path.realpath(__file__)).split("\webapp")[0]
|
| 16 |
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
|
|
|
|
| 129 |
return clause2labels
|
| 130 |
|
| 131 |
|
|
|
|
| 132 |
def run_pipeline(text):
|
| 133 |
snippets = auto_split(text)
|
| 134 |
print(snippets)
|