diff --git a/main.py b/main.py
index 8ce0146..cefa8e3 100755
--- a/main.py
+++ b/main.py
@@ -1,5 +1,14 @@
 import webui as mrq
 
+if 'XDG_CACHE_HOME' not in os.environ:
+    os.environ['XDG_CACHE_HOME'] = os.path.realpath('./models/')
+
+if 'TORTOISE_MODELS_DIR' not in os.environ:
+    os.environ['TORTOISE_MODELS_DIR'] = os.path.realpath('./models/tortoise/')
+
+if 'TRANSFORMERS_CACHE' not in os.environ:
+    os.environ['TRANSFORMERS_CACHE'] = os.path.realpath('./models/transformers/')
+
 if __name__ == "__main__":
     mrq.args = mrq.setup_args()
 
diff --git a/tortoise/api.py b/tortoise/api.py
index 7ca0a74..cc006a9 100755
--- a/tortoise/api.py
+++ b/tortoise/api.py
@@ -6,12 +6,6 @@ import gc
 from time import time
 from urllib import request
 
-if 'TORTOISE_MODELS_DIR' not in os.environ:
-    os.environ['TORTOISE_MODELS_DIR'] = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../models/tortoise/')
-
-if 'TRANSFORMERS_CACHE' not in os.environ:
-    os.environ['TRANSFORMERS_CACHE'] = os.path.join(os.path.dirname(os.path.realpath(__file__)), '../models/transformers/')
-
 import torch
 import torch.nn.functional as F
 import progressbar
@@ -38,7 +32,7 @@ pbar = None
 
 STOP_SIGNAL = False
 
-MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR')
+MODELS_DIR = os.environ.get('TORTOISE_MODELS_DIR', os.path.realpath('./models/tortoise/'))
 MODELS = {
     'autoregressive.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/autoregressive.pth',
     'classifier.pth': 'https://huggingface.co/jbetker/tortoise-tts-v2/resolve/main/.models/classifier.pth',