Framby commited on
Commit
166de0f
·
verified ·
1 Parent(s): ad24bb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -17,9 +17,8 @@ from huggingface_hub import hf_hub_download
17
  st.set_page_config(page_title="Tag Predictor", layout="wide")
18
 
19
  REPO_ID = "Framby/deberta_multilabel"
20
- deberta_path = hf_hub_download(
21
- repo_id=REPO_ID, filename="deberta_multilabel.pt")
22
- mlb_path = hf_hub_download(repo_id=REPO_ID, filename="mlb.pkl")
23
 
24
 
25
  @st.cache_resource
 
17
  st.set_page_config(page_title="Tag Predictor", layout="wide")
18
 
19
  REPO_ID = "Framby/deberta_multilabel"
20
+ mlb_path = hf_hub_download(repo_id=REPO_ID, filename="mlb.joblib")
21
+ mlb = joblib.load(mlb_path)
 
22
 
23
 
24
  @st.cache_resource