INNER CODE UNIT · Python

HousingData

balapriyac/data-science-tutorials · deploy_ml_models/app.py:14

class HousingData(BaseModel):
    MedInc: float
    HouseAge: float
    AveRooms: float
    AveOccup: float

# Define the prediction endpoint
@app.post('/predict')
def predict_price(data: HousingData):
    # Convert input data to a DataFrame
    input_data = pd.DataFrame([data.dict()])
    
    # Make a prediction
    prediction = model.predict(input_data)
    
    # Return the predicted price
    return {"Predicted Price": prediction[0]}

View source record →