Upload 2 files
Browse files
src/prediction_compile.py
CHANGED
|
@@ -47,8 +47,8 @@ def load_tokenizer_params():
|
|
| 47 |
|
| 48 |
@st.cache_resource
|
| 49 |
def load_topic_models():
|
| 50 |
-
neg_path = "./src/
|
| 51 |
-
pos_path = "./src/
|
| 52 |
neg_model = joblib.load(neg_path)
|
| 53 |
pos_model = joblib.load(pos_path)
|
| 54 |
return neg_model, pos_model
|
|
|
|
| 47 |
|
| 48 |
@st.cache_resource
|
| 49 |
def load_topic_models():
|
| 50 |
+
neg_path = "./src/fastopic_negative_model_10.pkl"
|
| 51 |
+
pos_path = "./src/fastopic_positive_model_10.pkl"
|
| 52 |
neg_model = joblib.load(neg_path)
|
| 53 |
pos_model = joblib.load(pos_path)
|
| 54 |
return neg_model, pos_model
|