Aiden McComiskey commited on
Commit
8d89728
Β·
1 Parent(s): 56f1c8f

Changing the model to de-duped

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  import joblib, numpy as np, torch, transformers
3
 
4
  # ── load artefacts once ───────────────────────────────────────────────
5
- rf = joblib.load("models/ovr_rf_bge.joblib")
6
- thr = np.load("models/thresholds_bge.npy")
7
  mlb = joblib.load("models/mlb.joblib")
8
 
9
  embedder = transformers.AutoModel.from_pretrained(
 
2
  import joblib, numpy as np, torch, transformers
3
 
4
  # ── load artefacts once ───────────────────────────────────────────────
5
+ rf = joblib.load("models/RandomForest_tfidf_pipe.joblib")
6
+ thr = np.load("models/RandomForest_tfidf_thresh.npy")
7
  mlb = joblib.load("models/mlb.joblib")
8
 
9
  embedder = transformers.AutoModel.from_pretrained(