maple-shaft commited on
Commit
064894c
·
verified ·
1 Parent(s): 17d2184

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -98,7 +98,7 @@ class HFMultiViewGen:
98
  class EndpointHandler():
99
  def __init__(self, path=""):
100
  self.hf_token = os.environ["HUGGINGFACE_TOKEN"]
101
- self.repo_dir = os.environ["HF_HUB_CACHE"]
102
  self.hf_gen = HFMultiViewGen(hf_token=self.hf_token, repo_dir=self.repo_dir)
103
 
104
  def convert(self, fromval: dict[str, Image.Image]) -> dict[str, str]:
 
98
  class EndpointHandler():
99
  def __init__(self, path=""):
100
  self.hf_token = os.environ["HUGGINGFACE_TOKEN"]
101
+ self.repo_dir = os.environ["HF_HUB_CACHE"] if not path else path
102
  self.hf_gen = HFMultiViewGen(hf_token=self.hf_token, repo_dir=self.repo_dir)
103
 
104
  def convert(self, fromval: dict[str, Image.Image]) -> dict[str, str]: