Spaces:
Sleeping
Sleeping
Update tasks/utils/predict.py
Browse files- tasks/utils/predict.py +2 -0
tasks/utils/predict.py
CHANGED
|
@@ -2,6 +2,8 @@ import pickle
|
|
| 2 |
import re
|
| 3 |
import string
|
| 4 |
import pandas as pd
|
|
|
|
|
|
|
| 5 |
from preprocessing import process_text
|
| 6 |
|
| 7 |
def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str):
|
|
|
|
| 2 |
import re
|
| 3 |
import string
|
| 4 |
import pandas as pd
|
| 5 |
+
import sys
|
| 6 |
+
sys.path.append(".")
|
| 7 |
from preprocessing import process_text
|
| 8 |
|
| 9 |
def predict(input_df: pd.DataFrame, tfidf_path: str, model_path: str):
|