forcing torch.load to CPU

app.py

@@ -2,7 +2,12 @@ import gradio as gr
 from pathlib import Path
 
 import soundfile as sf
+
+# forcing torch.load to CPU
 import torch
+_old_load = torch.load
+torch.load = lambda *a, **k: _old_load(*a, **{**k, 'map_location': 'cpu'})
+
 import torchaudio
 import hydra
 from omegaconf import OmegaConf