Hugging Face's logo

Spaces:
adrian7305
/
email-classifier-api
Sleeping

App Files Community
adrian7305 commited on
Commit
f0e7ec4
·
1 Parent(s): c0f1317

Fix permission

Browse files
Files changed (1) hide show
  1. models.py +5 -1
models.py CHANGED
@@ -8,7 +8,11 @@ MODEL_REPO = "adrian7305/email-classifier"
8
  MODEL_FILE = "model.joblib"
9
 
10
  def load_model() -> Pipeline:
11
- model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 
 
 
 
12
  return load(model_path)
13
 
14
  def classify_email(model: Pipeline, email: str) -> str:
 
8
  MODEL_FILE = "model.joblib"
9
 
10
  def load_model() -> Pipeline:
11
+ model_path = hf_hub_download(
12
+ repo_id=MODEL_REPO,
13
+ filename=MODEL_FILE,
14
+ cache_dir="./.hf_cache"
15
+ )
16
  return load(model_path)
17
 
18
  def classify_email(model: Pipeline, email: str) -> str: