Update gradio_app.py

gradio_app.py

@@ -5,8 +5,7 @@ import gradio as gr
 import look2hear.models
 
 # Setup environment and model
-os.environ["CUDA_VISIBLE_DEVICES"] = "2"
-device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 model = look2hear.models.TIGERDNR.from_pretrained("JusperLee/TIGER-DnR", cache_dir="cache")
 model.to(device)