Update app.py

app.py

@@ -6,10 +6,12 @@ import numpy as np
 import gradio as gr
 
 def _grab_best_device(use_gpu=True):
-    if torch.cuda.device_count() > 0 and use_gpu:
-        device = "cuda"
-    else:
-        device = "cpu"
+    #if torch.cuda.device_count() > 0 and use_gpu:
+    #    device = "cuda"
+    #else:
+    #    device = "cpu"
+    device = 0 if torch.cuda.is_available() else -1
+    
     return device
 
 device = _grab_best_device()
@@ -41,7 +43,7 @@ HUB_PATH = "ylacombe/vits_ljs_midlands_male_monospeaker"
 
 pipe_dict = {
     "current_model": "ylacombe/vits_ljs_midlands_male_monospeaker",
-    "pipe":  pipeline("text-to-speech", model=HUB_PATH, device=0),
+    "pipe":  pipeline("text-to-speech", model=HUB_PATH, device=device),
     "original_pipe": pipeline("text-to-speech", model=default_model_per_language["english"], device=0),
     "language": "english",
 }