Fix weights loading on CPU + requirements.txt + fix example command

[AlexMili]

Hello,

Thank you for your great work.

This PR is just to add a fix on weights loading when on CPU. Since the weights were obtained from a GPU, it is necessary to specify the map_location argument to convert it to CPU (details on torch website's here). I also added a requirements.txt to facilitate installation.