LPX commited on
Commit
d7aaf8c
·
1 Parent(s): e8835e7

fix: store full result dictionary in model predictions for enhanced data tracking

Browse files
Files changed (1) hide show
  1. app_mcp.py +1 -1
app_mcp.py CHANGED
@@ -292,7 +292,7 @@ def predict_image_with_json(img, confidence_threshold, augment_methods, rotate_d
292
  model_end - model_start
293
  )
294
 
295
- model_predictions_raw[model_id] = result["Label"]
296
  confidence_scores[model_id] = max(result.get("AI Score", 0.0), result.get("Real Score", 0.0))
297
  results.append(result) # Add individual model result to the list
298
 
 
292
  model_end - model_start
293
  )
294
 
295
+ model_predictions_raw[model_id] = result # Store the full result dictionary
296
  confidence_scores[model_id] = max(result.get("AI Score", 0.0), result.get("Real Score", 0.0))
297
  results.append(result) # Add individual model result to the list
298