Update app.py

app.py

@@ -1,5 +1,3 @@
-import os
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
 from utils.hparams import hparams
 from preprocessing.data_gen_utils import get_pitch_parselmouth,get_pitch_crepe
 import numpy as np
@@ -11,7 +9,7 @@ import torchcrepe
 from infer import *
 import logging
 from infer_tools.infer_tool import *
-map_location = torch.device("cpu")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
 ##EDIT FOR CPU
 # Open the file and read it into a string