CyberWaifu commited on
Commit
cfb04b7
·
verified ·
1 Parent(s): 677217d

Change default threshold to overall micro_opt

Browse files

More thresholds to try here: https://huggingface.co./Camais03/camie-tagger/blob/main/model/thresholds.json

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ import torchvision.transforms as transforms
10
  MODEL_REPO = "AngelBottomless/camie-tagger-onnxruntime"
11
  MODEL_FILE = "camie_tagger_initial.onnx"
12
  META_FILE = "metadata.json"
13
- DEFAULT_THRESHOLD = 0.35 # Default value if slider is not used
14
 
15
  # Download model and metadata from Hugging Face Hub
16
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE, cache_dir=".")
 
10
  MODEL_REPO = "AngelBottomless/camie-tagger-onnxruntime"
11
  MODEL_FILE = "camie_tagger_initial.onnx"
12
  META_FILE = "metadata.json"
13
+ DEFAULT_THRESHOLD = 0.32626262307167053 # Default value if slider is not used
14
 
15
  # Download model and metadata from Hugging Face Hub
16
  model_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE, cache_dir=".")