sheikh987 commited on
Commit
ba3cd47
·
verified ·
1 Parent(s): cd939e3

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -137,7 +137,7 @@ async def analyze_image(request: ImageRequest):
137
  confidence, predicted_idx = torch.max(probabilities, 1)
138
  confidence_percent = confidence.item() * 100
139
 
140
- if confidence_percent < 60.0:
141
  return {"status": "Inconclusive", "message": f"Model confidence ({confidence_percent:.2f}%) is below the 75% threshold."}
142
 
143
  # Stage 3: Return result
 
137
  confidence, predicted_idx = torch.max(probabilities, 1)
138
  confidence_percent = confidence.item() * 100
139
 
140
+ if confidence_percent < 50.0:
141
  return {"status": "Inconclusive", "message": f"Model confidence ({confidence_percent:.2f}%) is below the 75% threshold."}
142
 
143
  # Stage 3: Return result