Update app.py

app.py

@@ -122,7 +122,10 @@ def Text_to_Speech(language):
     elif language=='Spanish':
         to_code = "es"
 
-    
+    # Get device
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    #Get tts model from TTS
+    tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
     #Create directory
     if os.path.exists('TTS'):
       shutil.rmtree('TTS')
@@ -166,10 +169,7 @@ def Text_to_Speech(language):
     return gr.Audio(value="TTS/TTS.wav"), gr.File(label="TTS Audio File", value="TTS/TTS.wav")
 
 with gr.Blocks() as demo:
-    # Get device
-    device = "cuda" if torch.cuda.is_available() else "cpu"
-    #Get tts model from TTS
-    tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2").to(device)
+    
     gr.Markdown(
     """
     # Foreign Whispers