Update app.py

app.py

@@ -38,31 +38,28 @@ def tts(text: str, voice: str = "Male", use_onnx: bool = True):
     if len(text) > MAX_TXT_LEN:
         text = text[:MAX_TXT_LEN]
         print(f"Input text was cutoff since it went over the {MAX_TXT_LEN} character limit.")
-    print(text)
+
+    model_dir = BASE_DIR.format("male" if voice == "Male" else "female")
+    config_file = f"{model_dir}/config.json"
 
     text = text.replace("I", "ӏ")  # Replace capital "I" with "Palochka" symbol
     text = text.lower()
 
     if use_onnx:
-        # Load the ONNX model
         onnx_model_url = MALE_ONNX_MODEL_URL if voice == "Male" else FEMALE_ONNX_MODEL_URL
 
         config = VitsConfig()
-        config.load_json("config.json")
+        config.load_json(config_file)
 
-        # Initialize the tokenizer
         tokenizer = TTSTokenizer(
             use_phonemes=False,
             text_cleaner=config.text_cleaner,
-            characters=VitsCharacters(),  # Assuming the config has character info
+            characters=VitsCharacters(),
             phonemizer=None,
             add_blank=config.add_blank,
         )
 
-        # Create the Vits model instance
         vits = Vits.init_from_config(config)
-
-        # Load the ONNX model into the Vits model
         vits.load_onnx(onnx_model_url)
 
         text_inputs = np.asarray(
@@ -76,10 +73,9 @@ def tts(text: str, voice: str = "Male", use_onnx: bool = True):
             out_path = temp_file.name
         save_wav(wav=audio[0], path=out_path, sample_rate=24000)
     else:
-        model_dir = BASE_DIR.format("male" if voice == "Male" else "female")
 
         # Synthesize
-        synthesizer = Synthesizer(f"{model_dir}/model.pth", f"{model_dir}/config.json")
+        synthesizer = Synthesizer(f"{model_dir}/model.pth", config_file)
         wavs = synthesizer.tts(text)
 
         # Create a temporary WAV file