Update app.py

app.py

@@ -1,6 +1,5 @@
 import torch
 import os
-#os.system('pip install git+https://github.com/peterwilli/audio-maister.git')
 import gradio as gr
 
 from audiomaister import VoiceFixer
@@ -20,7 +19,7 @@ def load_default_weights():
 
 def inference(input_file, **kwargs):
     checkpoint = load_default_weights()
-    state = torch.load(checkpoint)
+    state = torch.load(checkpoint, map_location=torch.device('cuda' if USE_CUDA else 'cpu'))
     
     main_model = VoiceFixer(state['hparams'], 1, 'vocals')
     main_model.load_state_dict(state['weights'])