Lack of function to reproduce result

Hi the authors,

Thanks for making such a great work and i would like to apply this method into my current work. Correct me if i'm wrong but when I try to run `BayesCap_SRGAN_train_and_eval.ipynb`, it appears that you might lack the function of `eval_BayesCap` to fully reproduce the complete functionalities of the notebook itself. 

Could you kindly help me with this problem. Thanks a lot