Hey @akia , I couldn’t reply anymore to the topic because I’m a new user with a cap on replies. Couldn’t PM you either. So posting a new topic for now.
I still think your out-of-memory error is due to the size of the dataframe that’s being sent collected back by toPandas().
You could also verify that it’s not the Plotly and Zeppelin components leaking memory by trying the sample notebook I included in my repository. That one creates several visualizations and whether or not it’s able to run in your environment might help us troubleshoot your problems.
Cheers.