Update main.py
Browse files
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 <
|
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
|