Hello! A requirements.txt could be very useful for those trying to run the code. Also, specifications on GPU requirements seem to be missing. Thanks!