LovnishVerma commited on
Commit
ae5c3b8
·
verified ·
1 Parent(s): 95706ff

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -66,7 +66,7 @@ def resultbt():
66
  confidence = prediction if prediction > 0.5 else 1 - prediction # Calculate confidence
67
  predicted_class = 'Tumor Detected' if prediction > 0.5 else 'No Tumor Detected' # Determine class based on threshold
68
 
69
- # Prepare data for MongoDB
70
  result = {
71
  "firstname": firstname,
72
  "lastname": lastname,
@@ -76,7 +76,7 @@ def resultbt():
76
  "age": age,
77
  "image_name": filename,
78
  "prediction": predicted_class,
79
- "confidence_score": confidence,
80
  "timestamp": datetime.utcnow()
81
  }
82
 
 
66
  confidence = prediction if prediction > 0.5 else 1 - prediction # Calculate confidence
67
  predicted_class = 'Tumor Detected' if prediction > 0.5 else 'No Tumor Detected' # Determine class based on threshold
68
 
69
+ # Prepare data for MongoDB with JSON-serializable fields
70
  result = {
71
  "firstname": firstname,
72
  "lastname": lastname,
 
76
  "age": age,
77
  "image_name": filename,
78
  "prediction": predicted_class,
79
+ "confidence_score": float(confidence), # Ensure it's a standard float for JSON serialization
80
  "timestamp": datetime.utcnow()
81
  }
82