I get the same issue as you mentioned. I write a python script for loading model and run it in jupyter:
%run load_model.py
@z3ppj1g5aw Although it’s not the proper solution, but could be considered as a remedy.