Update demo.py: fix inference OOM error

Hi, thanks for open-sourcing such wonderful work! I noticed that once the number of inference audio increases, it will run into the OOM error because all the predictions are saved in GPU. The solution is to move the predictions to cpu as `out_dict[wav_f][dataset_name] = out.cpu()`.