Rthur2003 commited on
Commit
a55f921
·
1 Parent(s): f999d90

fix: adjust confidence calculation and streamline model download logic

Browse files
Files changed (2) hide show
  1. app/routes/analyze.py +1 -1
  2. startup.py +3 -5
app/routes/analyze.py CHANGED
@@ -516,7 +516,7 @@ async def _analyze_file(
516
  final_decision = fusion.decision_source
517
  if xai_payload is not None:
518
  final_is_ai = xai_payload.probability >= xai_payload.threshold
519
- final_confidence = round(xai_payload.probability, 4)
520
  final_decision = f"auris_xai_{xai_payload.bestModel.lower()}"
521
 
522
  result = AnalysisResult(
 
516
  final_decision = fusion.decision_source
517
  if xai_payload is not None:
518
  final_is_ai = xai_payload.probability >= xai_payload.threshold
519
+ final_confidence = round(max(0.03, min(0.97, xai_payload.probability)), 4)
520
  final_decision = f"auris_xai_{xai_payload.bestModel.lower()}"
521
 
522
  result = AnalysisResult(
startup.py CHANGED
@@ -104,12 +104,10 @@ def download_models() -> None:
104
  print(f"[startup] skip {filename} (exists)")
105
  continue
106
 
107
- if filename not in repo_files and filename not in REQUIRED_FILES:
108
- if filename in OPTIONAL_FILES:
109
- print(f"[startup] skip {filename} (not in repo, optional)")
110
- continue
111
-
112
  is_optional = filename in OPTIONAL_FILES
 
 
 
113
  try:
114
  print(f"[startup] dl {filename} ...", end=" ", flush=True)
115
  path = hf_hub_download(
 
104
  print(f"[startup] skip {filename} (exists)")
105
  continue
106
 
 
 
 
 
 
107
  is_optional = filename in OPTIONAL_FILES
108
+ if filename not in repo_files and is_optional:
109
+ print(f"[startup] skip {filename} (not in repo, optional)")
110
+ continue
111
  try:
112
  print(f"[startup] dl {filename} ...", end=" ", flush=True)
113
  path = hf_hub_download(