Hello, Tudosiu Thank you for sharing your great work. I think the shared weight file is broken since the load function yields the following error message. Could you check that the files are correct? Thank you in advance. Suhyun.  