haiya

Dockerfile

@@ -10,4 +10,6 @@ COPY . .
 
 RUN mkdir ./.cache
 
+ENV HF_DATASETS_CACHE="./.cache"
+
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

hugging_face/dataset.py

@@ -7,7 +7,12 @@ def load_dataset(filename) -> DataFrame:
   try:
     df = read_csv(filename)
   except:
-    datasets = hf_load_dataset(REPO_ID, data_files=filename, encoding='latin-1')
+    datasets = hf_load_dataset(
+      REPO_ID, 
+      data_files=filename, 
+      encoding='latin-1'
+    )
+
     df = DataFrame(data=datasets['train'])
     df.to_csv(filename, index=False)
 

hugging_face/model.py

@@ -5,5 +5,9 @@ REPO_ID = "kompiangg/svm-hate-speech-classification"
 
 def load_hugging_face_model(filename):
   return joblib.load(
-    hf_hub_download(REPO_ID, filename, local_dir="./models")
+    hf_hub_download(
+      REPO_ID, 
+      filename, 
+      local_dir="./models", 
+    )
   )