Unearthed commited on
Commit
f1b6f98
Β·
verified Β·
1 Parent(s): 756df06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ def display_category(cls_name):
43
  # ── Load model at startup ─────────────────────────────────────
44
  print("Loading model...")
45
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 
 
46
  model = tf.keras.models.load_model(model_path, compile=False)
47
  print("Model ready!")
48
 
 
43
  # ── Load model at startup ─────────────────────────────────────
44
  print("Loading model...")
45
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
46
+ import keras
47
+ keras.config.enable_unsafe_deserialization()
48
  model = tf.keras.models.load_model(model_path, compile=False)
49
  print("Model ready!")
50