Update xtts.py

xtts.py

@@ -39,18 +39,18 @@ def predict(text, sample=None, language="zh"):
     global model
     try:
         if tts is None:
-            # model_dir=os.environ.get("MODEL_DIR")
-            # model_path=model_dir
-            # config_path=f'{model_dir}/config.json'
-            # vocoder_config_path=f'{model_dir}/vocab.json'
+            model_dir=os.environ.get("MODEL_DIR")
+            model_path=model_dir
+            config_path=f'{model_dir}/config.json'
+            vocoder_config_path=f'{model_dir}/vocab.json'
             model_name="tts_models/multilingual/multi-dataset/xtts_v2"
             logging.info(f"loading model {model_name} ...")
             tts = TTS(
-                model_name,
-                # model_path=model_path, 
-                # config_path=config_path, 
-                # vocoder_config_path=vocoder_config_path,  
-                progress_bar=False
+                # model_name,
+                model_path=model_path, 
+                config_path=config_path, 
+                vocoder_config_path=vocoder_config_path,  
+                progress_bar=True
             )
             model=tts.synthesizer.tts_model
             #hack to use cache