Update xtts.py

xtts.py

@@ -10,14 +10,18 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 tts=None
 model=None 
 
-sample_root= os.environ.get('XTTS_SAMPLE_DIR') 
-if(sample_root==None):
-    sample_root=f'{os.getcwd()}/samples'
+root=os.path.dirname(os.path.abspath(__file__))
+sample_root=f'{root}/samples'
 if not os.path.exists(sample_root):
     os.makedirs(sample_root)
+    
+LIBROSA_CACHE_DIR=f'{root}/caches'
+if not os.path.exists(LIBROSA_CACHE_DIR):
+    os.makedirs(LIBROSA_CACHE_DIR)
+os.environ["LIBROSA_CACHE_DIR"]=LIBROSA_CACHE_DIR
 
-default_sample=f'{os.path.dirname(os.path.abspath(__file__))}/sample.wav', f'{sample_root}/sample.pt'
-ffmpeg=f'{os.path.dirname(os.path.abspath(__file__))}/ffmpeg'
+default_sample=f'{root}/sample.wav', f'{sample_root}/sample.pt'
+ffmpeg=f'{root}/ffmpeg'
 
 if api:
     from qili import upload
@@ -157,11 +161,10 @@ def tts_play():
 
 
 @app.route("/setup")
-def get_tts(bLOCAL=None):
+def get_tts(model_path=os.environ["MODEL_DIR"]):
     global tts 
     global model
     if tts is None:
-        model_path=os.environ.get("MODEL_DIR")
         config_path=f'{model_path}/config.json'
         vocoder_config_path=f'{model_path}/vocab.json'
         model_name="tts_models/multilingual/multi-dataset/xtts_v2"