Spaces:
Sleeping
Sleeping
Aiden McComiskey
commited on
Commit
Β·
8d89728
1
Parent(s):
56f1c8f
Changing the model to de-duped
Browse files
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/
|
| 6 |
-
thr = np.load("models/
|
| 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(
|