Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 app = FastAPI()
 
 class InputData(BaseModel):
-    input1 : bool
+    input1 : float
     input2 : float
     input3 : float
     input4 : float
@@ -31,4 +31,4 @@ def predict(input : InputData):
     input.input5, input.input6,
     input.input7]])
     prediction = model.predict(data).tolist()
-    return {'prediction' : f'prediction[0]'}
+    return {'prediction' : prediction[0]}