Update main.py

main.py

@@ -55,6 +55,6 @@ async def predict(file: UploadFile = File(...)):
     
     # Check if the predicted probability meets the class-specific threshold
     if predicted_probability < thresholds[predicted_idx]:
-        return {'predicted_class': 'uncertain', 'accuracy': predicted_probability}
+        return {'predicted_class': 'uncertain', 'accuracy': float(predicted_probability)}
     else:
-        return {'predicted_class': predicted_label, 'accuracy': predicted_probability}
+        return {'predicted_class': predicted_label, 'accuracy': float(predicted_probability)}