Spaces:

alvanlii
/

domain-expansion

Runtime error

alvanlii commited on Feb 15, 2023

Commit

6d721db

1 Parent(s): 70c3221

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,10 @@
 # Copyright 2023 Adobe Research. All rights reserved.
 # To view a copy of the license, visit LICENSE.md.
 import os
-os.environ["HF_DATASETS_CACHE="] = "/exp/domain-expansion/.cache"
 import torch

 # Copyright 2023 Adobe Research. All rights reserved.
 # To view a copy of the license, visit LICENSE.md.
 import os
+CACHE_DIR = "/exp/domain-expansion/.cache"
+os.environ["HF_DATASETS_CACHE"] = CACHE_DIR
+if not os.path.exists(CACHE_DIR):
+    os.mkdir(CACHE_DIR)
 import torch