Update load_pipeline_from_hub function

app.py

@@ -8,7 +8,7 @@ import datetime
 import numpy as np
 from huggingface_hub import hf_hub_download
 import torch
-
+import os
 
 pd.set_option('display.max_colwidth', 1000)
 
@@ -21,9 +21,16 @@ def cpu_load(*args, **kwargs):
 torch.load = cpu_load
 
 def load_pipeline_from_hub(filename):
+    cache_dir = "/tmp/hf_cache"
+    os.environ["HF_HUB_CACHE"] = cache_dir  # optional but informative
+
     repo_id = 'hw01558/nlp-coursework-pipelines'
-    local_path = hf_hub_download(repo_id=repo_id, filename=filename)
+    local_path = hf_hub_download(repo_id=repo_id, filename=filename, cache_dir=cache_dir)
     return load(local_path)
+    
+    #repo_id = 'hw01558/nlp-coursework-pipelines'
+    #local_path = hf_hub_download(repo_id=repo_id, filename=filename)
+    #return load(local_path)
 
 PIPELINES = [
     {