HelloNIKHIL KUMAR
"Driver restarted - possibly due to an OutOfMemoryError" indicates the driver node in your Databricks cluster has run out of memory.
This can be due to insufficient memory allocation on the driver node or inefficient code.
Did you see any exception messages in the driver logs?
Could you please try the below?
Can you edit the Databricks cluster configuration and increase the driver node memory allocation?
And if you use a smaller cluster, please try with a different size(large cluster) and see if it helps.
also, see if you can optimize your code
I hope this helps.