Update handler.py
Browse files- 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]:
|