DraconicDragon commited on
Commit
ca53484
·
verified ·
1 Parent(s): 1170578

add version 1_01 of model, probably, hopefully

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -274,8 +274,9 @@ def visualize_predictions(image: Image.Image, predictions: Dict, threshold: floa
274
  REPO_ID = "cella110n/cl_tagger"
275
  MODEL_OPTIONS = {
276
  "cl_tagger_1_00": "cl_tagger_1_00/model_optimized.onnx"
 
277
  }
278
- DEFAULT_MODEL = "cl_tagger_1_00"
279
  CACHE_DIR = "./model_cache"
280
 
281
  # --- Global variables for paths (initialized at startup) ---
 
274
  REPO_ID = "cella110n/cl_tagger"
275
  MODEL_OPTIONS = {
276
  "cl_tagger_1_00": "cl_tagger_1_00/model_optimized.onnx"
277
+ "cl_tagger_1_01": "cl_tagger_1_01/model_optimized.onnx"
278
  }
279
+ DEFAULT_MODEL = "cl_tagger_1_01"
280
  CACHE_DIR = "./model_cache"
281
 
282
  # --- Global variables for paths (initialized at startup) ---