Spaces:
Sleeping
Sleeping
marintosti12 commited on
Commit ·
09ccbd8
1
Parent(s): 84394db
fix(loader) : fix env loader
Browse files- .github/workflows/ci.yaml +2 -0
- src/model_loader.py +1 -1
.github/workflows/ci.yaml
CHANGED
|
@@ -18,6 +18,8 @@ jobs:
|
|
| 18 |
POSTGRES_USER: ci
|
| 19 |
POSTGRES_PASSWORD: ci
|
| 20 |
POSTGRES_DB: futurisys_ci
|
|
|
|
|
|
|
| 21 |
ports:
|
| 22 |
- 5432:5432
|
| 23 |
options: >-
|
|
|
|
| 18 |
POSTGRES_USER: ci
|
| 19 |
POSTGRES_PASSWORD: ci
|
| 20 |
POSTGRES_DB: futurisys_ci
|
| 21 |
+
APP_ENV: test
|
| 22 |
+
|
| 23 |
ports:
|
| 24 |
- 5432:5432
|
| 25 |
options: >-
|
src/model_loader.py
CHANGED
|
@@ -22,7 +22,7 @@ def _load_local(name: str) -> Any:
|
|
| 22 |
@lru_cache(maxsize=1)
|
| 23 |
def load_model(name) -> Any:
|
| 24 |
|
| 25 |
-
if ENV in ("dev"
|
| 26 |
return _load_local(name)
|
| 27 |
|
| 28 |
hf_path = hf_hub_download(
|
|
|
|
| 22 |
@lru_cache(maxsize=1)
|
| 23 |
def load_model(name) -> Any:
|
| 24 |
|
| 25 |
+
if ENV in ("dev"):
|
| 26 |
return _load_local(name)
|
| 27 |
|
| 28 |
hf_path = hf_hub_download(
|