Upload app.py

app.py

@@ -81,15 +81,15 @@ def predict_insurance_charge(age, bmi, children, sex, smoker, region):
                     'sex': sex,
                     'smoker': smoker,
                     'region': region,
-                    'prediction': prediction[0][0]
+                    'prediction': prediction[0]
                 }
             ))
             f.write("\n")
 
-    prediction_result = prediction[0][0]
+    prediction_result = prediction[0]
     print(f"\nDebug - Prediction result: {prediction_result} - {type(prediction_result)}\n")
-    print(f"\nDebug - Prediction result: {prediction_result[0]} - {type(prediction_result[0])}\n")
-    print(f"\nDebug - Prediction result: {prediction_result[0][0]} - {type(prediction_result[0][0])}\n")    
+    print(f"\nDebug - Prediction result[0]S: {prediction_result[0]} - {type(prediction_result[0])}\n")
+    #print(f"\nDebug - Prediction result: {prediction_result[0][0]} - {type(prediction_result[0][0])}\n")    
     return prediction_result
     
     #return prediction[0]