hanantonio commited on
Commit
293f9ac
·
verified ·
1 Parent(s): 4a9a3c2

Upload 2 files

Browse files
Files changed (1) hide show
  1. src/prediction_compile.py +2 -2
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/fastopic_negative_model.pkl"
51
- pos_path = "./src/fastopic_positive_model.pkl"
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