Spaces:
Running
Running
Fix writable runtime storage path
Browse files
api.py
CHANGED
|
@@ -24,13 +24,15 @@ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
| 24 |
def default_runtime_root():
|
| 25 |
if os.path.isdir("/data"):
|
| 26 |
return os.path.join("/data", "sporalize_runtime")
|
| 27 |
-
return os.path.join(
|
| 28 |
|
| 29 |
|
| 30 |
RUNTIME_ROOT = os.environ.get("SPORALIZE_RUNTIME_DIR", default_runtime_root())
|
| 31 |
ASSETS_RUNTIME_ROOT = os.environ.get("SPORALIZE_ASSETS_DIR", os.path.join(RUNTIME_ROOT, "assets"))
|
| 32 |
WEIGHTS_RUNTIME_ROOT = os.environ.get("SPORALIZE_WEIGHTS_DIR", os.path.join(RUNTIME_ROOT, "weights"))
|
| 33 |
DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(CURRENT_DIR, "Storage")
|
|
|
|
|
|
|
| 34 |
STORAGE_ROOT = os.environ.get(
|
| 35 |
"SPORALIZE_STORAGE_DIR",
|
| 36 |
os.path.join("/data", "sporalize_storage") if os.path.isdir("/data") else DEFAULT_LOCAL_STORAGE_ROOT,
|
|
|
|
| 24 |
def default_runtime_root():
|
| 25 |
if os.path.isdir("/data"):
|
| 26 |
return os.path.join("/data", "sporalize_runtime")
|
| 27 |
+
return os.path.join(os.path.expanduser("~"), ".sporalize_runtime")
|
| 28 |
|
| 29 |
|
| 30 |
RUNTIME_ROOT = os.environ.get("SPORALIZE_RUNTIME_DIR", default_runtime_root())
|
| 31 |
ASSETS_RUNTIME_ROOT = os.environ.get("SPORALIZE_ASSETS_DIR", os.path.join(RUNTIME_ROOT, "assets"))
|
| 32 |
WEIGHTS_RUNTIME_ROOT = os.environ.get("SPORALIZE_WEIGHTS_DIR", os.path.join(RUNTIME_ROOT, "weights"))
|
| 33 |
DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(CURRENT_DIR, "Storage")
|
| 34 |
+
if not os.path.isdir("/data") and not os.access(CURRENT_DIR, os.W_OK):
|
| 35 |
+
DEFAULT_LOCAL_STORAGE_ROOT = os.path.join(RUNTIME_ROOT, "storage")
|
| 36 |
STORAGE_ROOT = os.environ.get(
|
| 37 |
"SPORALIZE_STORAGE_DIR",
|
| 38 |
os.path.join("/data", "sporalize_storage") if os.path.isdir("/data") else DEFAULT_LOCAL_STORAGE_ROOT,
|