Update xtts.py

xtts.py

@@ -13,6 +13,10 @@ import torch
 import torchaudio
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+model_dir=os.environ.get("MODEL_DIR")
+model_path=f'{model_dir}/model.pth'
+config_path=f'{model_dir}/config.json'
+vocoder_config_path=f'{model_dir}/vocab.json'
 
 app = Flask(__name__)
 # def upload_bytes(bytes, ext=".wav"):
@@ -55,7 +59,12 @@ def predict():
             TTS=import_module("TTS.api").TTS
             model_name="tts_models/multilingual/multi-dataset/xtts_v2"
             logging.info(f"loading model {model_name} ...")
-            tts = TTS(model_name=model_name, progress_bar=False)
+            tts = TTS(
+                model_path=model_path, 
+                config_path=config_path, 
+                vocoder_config_path=vocoder_config_path,  
+                progress_bar=False
+            )
             model=tts.synthesizer.tts_model
             #hack to use cache
             model.__get_conditioning_latents=model.get_conditioning_latents