Spaces:
Running on Zero
Running on Zero
Rawal Khirodkar commited on
Commit ·
d053090
1
Parent(s): c43d03d
Normal: enable full preload of all 4 sizes; default = 1B (shipped)
Browse files
app.py
CHANGED
|
@@ -54,7 +54,7 @@ NORMAL_MODELS = {
|
|
| 54 |
"config": os.path.join(CONFIGS_DIR, "sapiens2_5b_normal_metasim_render_people-1024x768.py"),
|
| 55 |
},
|
| 56 |
}
|
| 57 |
-
DEFAULT_SIZE = "
|
| 58 |
|
| 59 |
# v1 binary fg/bg TorchScript model.
|
| 60 |
FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
|
|
@@ -96,13 +96,10 @@ def _get_fg_model():
|
|
| 96 |
return _fg_model
|
| 97 |
|
| 98 |
|
| 99 |
-
|
| 100 |
-
|
| 101 |
-
|
| 102 |
-
_get_normal_model(DEFAULT_SIZE)
|
| 103 |
_get_fg_model()
|
| 104 |
-
# for _size in NORMAL_MODELS:
|
| 105 |
-
# _get_normal_model(_size)
|
| 106 |
print("[startup] ready.")
|
| 107 |
|
| 108 |
|
|
|
|
| 54 |
"config": os.path.join(CONFIGS_DIR, "sapiens2_5b_normal_metasim_render_people-1024x768.py"),
|
| 55 |
},
|
| 56 |
}
|
| 57 |
+
DEFAULT_SIZE = "1B"
|
| 58 |
|
| 59 |
# v1 binary fg/bg TorchScript model.
|
| 60 |
FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
|
|
|
|
| 96 |
return _fg_model
|
| 97 |
|
| 98 |
|
| 99 |
+
print("[startup] pre-loading all normal sizes + fg/bg ...")
|
| 100 |
+
for _size in NORMAL_MODELS:
|
| 101 |
+
_get_normal_model(_size)
|
|
|
|
| 102 |
_get_fg_model()
|
|
|
|
|
|
|
| 103 |
print("[startup] ready.")
|
| 104 |
|
| 105 |
|