Automatic device detection

app.py

@@ -15,9 +15,20 @@ model_name = "tts_models/multilingual/multi-dataset/xtts_v2"
 #print(m)
 m = model_name
 
-tts = TTS(model_name, gpu=False)
-tts.to("cpu") # no GPU or Amd
-#tts.to("cuda") # cuda only
+# Automatic device detection
+if torch.cuda.is_available():
+    # cuda only
+    device_type = "cuda"
+    device_selection = "cuda:0"
+    data_type = torch.float16
+else:
+    # no GPU or Amd
+    device_type = "cpu"
+    device_selection = "cpu"
+    data_type = torch.float32
+
+tts = TTS(model_name, gpu=torch.cuda.is_available())
+tts.to(device_type)
 
 
 def predict(prompt, language, audio_file_pth, mic_file_path, use_mic, agree, request: gr.Request):
@@ -73,7 +84,7 @@ def predict(prompt, language, audio_file_pth, mic_file_path, use_mic, agree, req
             )
         except RuntimeError as e :
             if "device-assert" in str(e):
-                # cannot do anything on cuda device side error, need tor estart
+                # cannot do anything on cuda device side error, need to restart
                 gr.Warning("Unhandled Exception encounter, please retry in a minute")
                 print("Cuda device-assert Runtime encountered need restart")
                 sys.exit("Exit due to cuda device-assert")
@@ -127,7 +138,7 @@ gr.Interface(
         gr.Textbox(
             label="Text Prompt",
             info="One or two sentences at a time is better",
-            value="Hello, World !, here is an example of light voice cloning. Try to upload your best audio samples quality",
+            value="Hello, World!, here is an example of light voice cloning. Try to upload your best audio samples quality",
         ),
         gr.Dropdown(
             label="Language",
@@ -170,8 +181,8 @@ gr.Interface(
         ),
     ],
     outputs=[
-        gr.Video(label="Waveform Visual"),
-        gr.Audio(label="Synthesised Audio"),
+        gr.Video(label="Waveform Visual", autoplay=True),
+        gr.Audio(label="Synthesised Audio", autoplay=False),
     ],
     title=title,
     description=description,