Incorrect Inference Results

[abhimanyuchadha96]

Hi there @henzler @madhawav
I cloned the repo as is and only made the changes mentioned in this pull request

I also got the weights as mentioned in the shell script.

I ran

But the results are horrible. I think maybe the weights were updated or some structure changed?
I have kept the versions as mentioned in the requirements.txt
Input:-

Outputs:-