Unearthed commited on
Commit
b06e788
Β·
verified Β·
1 Parent(s): 5b7cf82

Update app.py

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