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