Hugging Face's logo

Spaces:
kompiangg
/
hate-speech-classification
Sleeping

App Files Community
kompiangg commited on
Commit
94a2ff4
•
1 Parent(s): fdb0e37

naise

Browse files
Files changed (2) hide show
  1. .gitignore +2 -1
  2. hugging_face/model.py +3 -1
.gitignore CHANGED
@@ -1,3 +1,4 @@
1
  venv
2
  __pycache__
3
- *.csv
 
 
1
  venv
2
  __pycache__
3
+ *.csv
4
+ models/
hugging_face/model.py CHANGED
@@ -4,4 +4,6 @@ import joblib
4
  REPO_ID = "kompiangg/svm-hate-speech-classification"
5
 
6
  def load_hugging_face_model(filename):
7
- return joblib.load(hf_hub_download(REPO_ID, filename))
 
 
 
4
  REPO_ID = "kompiangg/svm-hate-speech-classification"
5
 
6
  def load_hugging_face_model(filename):
7
+ return joblib.load(
8
+ hf_hub_download(REPO_ID, filename, local_dir="./models")
9
+ )