float confidence

app.py

@@ -84,7 +84,7 @@ async def predict_disease(file: UploadFile = File(...)):
         # Validate file type
         
         # Save uploaded file temporarily
-        with tempfile.NamedTemporaryFile(suffix=".jpg", delete=False) as tmp:
+        with tempfile.NamedTemporaryFile(suffix=".jpeg", delete=False) as tmp:
             temp_path = tmp.name
             tmp.write(await file.read())
             tmp.flush()  # Ensure data is written
@@ -103,7 +103,7 @@ async def predict_disease(file: UploadFile = File(...)):
         # Predict
         prediction = model.predict(input_arr)
         result_index = np.argmax(prediction)
-        confidence = prediction[0][result_index]
+        confidence = float(prediction[0][result_index])
         disease_name = class_name[result_index]