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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -19,6 +19,8 @@ st.set_page_config(page_title="Tag Predictor", layout="wide")
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
 
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
+ deberta_path = hf_hub_download(
23
+ repo_id=REPO_ID, filename="deberta_multilabel.pt")
24
 
25
 
26
  @st.cache_resource