Fola-lad commited on
Commit
30642f0
·
1 Parent(s): cd96633

download model at runtime via hf_hub_download

Browse files
Files changed (1) hide show
  1. 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
- _MODEL_PATH,
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"