Was this ever resolved? This was last year and I still only can access the predict method outside of the databricks playground.