Spaces:
Running
Running
fix: adjust confidence calculation and streamline model download logic
Browse files- app/routes/analyze.py +1 -1
- 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(
|