Fix output parameters

app.py

@@ -29,58 +29,61 @@ 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):
-    if True:
-        if use_mic:
-            if mic_file_path is None:
-                gr.Warning("Please record your voice with Microphone, or uncheck Use Microphone to use reference audios")
-                return (
-                    None,
-                    None,
-                )
-            else:
-                speaker_wav = mic_file_path
+    if use_mic:
+        if mic_file_path is None:
+            gr.Warning("Please record your voice with Microphone, or uncheck Use Microphone to use reference audios")
+            return (
+                None,
+                None,
+                None,
+            )
         else:
-            speaker_wav = audio_file_pth
+            speaker_wav = mic_file_path
+    else:
+        speaker_wav = audio_file_pth
 
-        if len(prompt) < 2:
-            gr.Warning("Please give a longer prompt text")
-            return (
-                    None,
-                    None,
-                )
-        if len(prompt) > 50000:
-            gr.Warning("Text length limited to 50000 characters for this demo, please try shorter text")
-            return (
-                    None,
-                    None,
-                )  
-        try:
-            if language == "fr":
-                if m.find("your") != -1:
-                    language = "fr-fr"
-            if m.find("/fr/") != -1:
-                language = None
-            tts.tts_to_file(
-                text=prompt,
-                file_path="output.wav",
-                speaker_wav=speaker_wav,
-                language=language
+    if len(prompt) < 2:
+        gr.Warning("Please give a longer prompt text")
+        return (
+                None,
+                None,
+                None,
             )
-        except RuntimeError as e :
-            if "device-assert" in str(e):
-                # 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")
-            else:
-                raise e
-            
+    if len(prompt) > 50000:
+        gr.Warning("Text length limited to 50000 characters for this demo, please try shorter text")
         return (
-            gr.make_waveform(
-                audio="output.wav",
-            ),
-            "output.wav",
+                None,
+                None,
+                None,
+            )  
+    try:
+        if language == "fr":
+            if m.find("your") != -1:
+                language = "fr-fr"
+        if m.find("/fr/") != -1:
+            language = None
+        tts.tts_to_file(
+            text=prompt,
+            file_path="output.wav",
+            speaker_wav=speaker_wav,
+            language=language
         )
+    except RuntimeError as e :
+        if "device-assert" in str(e):
+            # 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")
+        else:
+            raise e
+        
+    return (
+        gr.make_waveform(
+            audio="output.wav",
+        ),
+        "output.wav",
+        None,
+    )
 
 with gr.Blocks() as interface:
     gr.HTML("Multi-language Text-to-Speech")