DelaliScratchwerk commited on
Commit
980f96f
·
verified ·
1 Parent(s): 2ff327f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -8,6 +8,11 @@ MARGIN_THRESHOLD = <value from script>
8
 
9
  MODEL_ID = "DelaliScratchwerk/text-period-setfit"
10
 
 
 
 
 
 
11
  # Load labels: try from model repo, else local labels.json in the Space
12
  try:
13
  labels_path = hf_hub_download(MODEL_ID, "labels.json")
@@ -15,11 +20,6 @@ try:
15
  except Exception:
16
  LABELS = json.load(open("labels.json"))
17
 
18
- # thresholds – tweak later with validation
19
- TOP_K = 3
20
- UNCERTAINTY_THRESHOLD = 0.42 # if top1 prob below this → "uncertain"
21
- MARGIN_THRESHOLD = 0.08 # or if (top1 - top2) < this → "uncertain"
22
-
23
  model = SetFitModel.from_pretrained(MODEL_ID)
24
 
25
  def format_evidence(ev):
 
8
 
9
  MODEL_ID = "DelaliScratchwerk/text-period-setfit"
10
 
11
+ # thresholds – tweak later with validation
12
+ TOP_K = 3
13
+ UNCERTAINTY_THRESHOLD = 0.42 # if top1 prob below this → "uncertain"
14
+ MARGIN_THRESHOLD = 0.08 # or if (top1 - top2) < this → "uncertain"
15
+
16
  # Load labels: try from model repo, else local labels.json in the Space
17
  try:
18
  labels_path = hf_hub_download(MODEL_ID, "labels.json")
 
20
  except Exception:
21
  LABELS = json.load(open("labels.json"))
22
 
 
 
 
 
 
23
  model = SetFitModel.from_pretrained(MODEL_ID)
24
 
25
  def format_evidence(ev):