Unearthed commited on
Commit
f5cc49b
Β·
verified Β·
1 Parent(s): af41ddb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -15,7 +15,7 @@ from scipy.spatial import cKDTree
15
 
16
  # ── Config ───────────────────────────────────────────────────
17
  MODEL_REPO = "Unearthed/artifact-classifier"
18
- MODEL_FILE = "lithic_classifier0601.keras"
19
  IMG_SIZE = (224, 224)
20
 
21
  CONF_THRESHOLD = 0.55
@@ -42,11 +42,8 @@ def display_category(cls_name):
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 ───────────────────────────────────────
 
15
 
16
  # ── Config ───────────────────────────────────────────────────
17
  MODEL_REPO = "Unearthed/artifact-classifier"
18
+ MODEL_FILE = "lithic_classifier_hf.keras"
19
  IMG_SIZE = (224, 224)
20
 
21
  CONF_THRESHOLD = 0.55
 
42
 
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
 
49
  # ── Feature extraction ───────────────────────────────────────