Update main.py

main.py

@@ -69,13 +69,13 @@ async def predict_student_performance_api(req: form2):
         input_data_reshaped = input_data_as_numpy_array.reshape(1,-1)
 
         # Perform the prediction
-        prediction = model_ML_DiabetesPrediction.predict(input_data_reshaped)
+        prediction = model_ML_DiabetesPrediction.predict(input_data_reshaped)[0]
+        prediction = int(prediction)
 
-        return JSONResponse(content={"ok": 1, "prediction": prediction[0]})
+        return JSONResponse(content={"ok": 1, "prediction": prediction})
     except Exception as e:
         return JSONResponse(content={"ok": -1, "message": f"Something went wrong! {str(e)}"}, status_code=500)
 
-
 # Main function to run the FastAPI server
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)