Spaces:
Runtime error
Runtime error
download model at runtime via hf_hub_download
Browse files- src/streamlit_app.py +8 -4
src/streamlit_app.py
CHANGED
|
@@ -33,14 +33,18 @@ EXPLANATIONS = {
|
|
| 33 |
|
| 34 |
@st.cache_resource
|
| 35 |
def load_model():
|
| 36 |
-
if not os.path.exists(_MODEL_PATH):
|
| 37 |
-
root_files = os.listdir(_REPO_ROOT)
|
| 38 |
-
return None, f"error: model.keras not found at {_MODEL_PATH}. Files in {_REPO_ROOT}: {root_files}"
|
| 39 |
try:
|
|
|
|
| 40 |
import tensorflow as tf
|
| 41 |
from model_def import FeedForwardNetwork
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
model = tf.keras.models.load_model(
|
| 43 |
-
|
| 44 |
custom_objects={"FeedForwardNetwork": FeedForwardNetwork},
|
| 45 |
)
|
| 46 |
return model, "ready"
|
|
|
|
| 33 |
|
| 34 |
@st.cache_resource
|
| 35 |
def load_model():
|
|
|
|
|
|
|
|
|
|
| 36 |
try:
|
| 37 |
+
from huggingface_hub import hf_hub_download
|
| 38 |
import tensorflow as tf
|
| 39 |
from model_def import FeedForwardNetwork
|
| 40 |
+
|
| 41 |
+
model_path = hf_hub_download(
|
| 42 |
+
repo_id="Group3DActRecog/actRecog",
|
| 43 |
+
filename="model.keras",
|
| 44 |
+
repo_type="space",
|
| 45 |
+
)
|
| 46 |
model = tf.keras.models.load_model(
|
| 47 |
+
model_path,
|
| 48 |
custom_objects={"FeedForwardNetwork": FeedForwardNetwork},
|
| 49 |
)
|
| 50 |
return model, "ready"
|