Update main.py

main.py

@@ -1,5 +1,5 @@
 # Import required libraries
-from fastapi import FastAPI, HTTPException, Form
+from fastapi import FastAPI, HTTPException, Form, Depends
 from pydantic import BaseModel
 import uvicorn
 import joblib
@@ -59,20 +59,10 @@ class Sepsis(BaseModel):
 
 # Define a route for prediction
 @app.post("/predict/")
-async def create_dataframe(form_data: Sepsis):
+async def create_dataframe(form_data: Sepsis = Depends(Sepsis.as_form)):
     try:
-        df = pd.DataFrame({
-    "plasma_glucose": int(form_data.plasma_glucose),
-    "blood_work_result_1": int(form_data.blood_work_result_1),
-    "blood_pressure": int(form_data.blood_pressure),
-    "blood_work_result_2": int(form_data.blood_work_result_2),
-    "blood_work_result_3": float(form_data.blood_work_result_3),
-    "body_mass_index": float(form_data.body_mass_index),
-    "blood_work_result_4": float(form_data.blood_work_result_4),
-    "Age": int(form_data.Age),
-    "Insurance": int(form_data.Insurance)
-}, index=[0])
-
+        # Convert the form data to a data frame
+        df = pd.DataFrame(form_data.dict(), index=[0])
 
         # Predicting
         output = model.predict_proba(df)