Hi @snarayan ,
I think you might be hitting timeout from model serving endpoint:
Debug model serving timeouts - Azure Databricks | Microsoft Learn
You can try to increase timeout using environment variables using the Serving UI or programmatically using Python.
