Unearthed commited on
Commit
1cbf676
Β·
verified Β·
1 Parent(s): 8a3838b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -38,6 +38,7 @@ def display_category(cls_name):
38
  # ── Load model ───────────────────────────────────────────────
39
  print("Loading model...")
40
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 
41
  model = tf.keras.models.load_model(model_path, compile=False)
42
  print("Model ready!")
43
 
 
38
  # ── Load model ───────────────────────────────────────────────
39
  print("Loading model...")
40
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
41
+ tf.keras.config.enable_unsafe_deserialization()
42
  model = tf.keras.models.load_model(model_path, compile=False)
43
  print("Model ready!")
44