nakany
Databricks Partner

data_json in the score_model function should be defined as follows

ds_dict = {"dataframe_split": dataset.to_dict(orient='split')} if isinstance(dataset, pd.DataFrame) else create_tf_serving_json(dataset)