djkesu commited on
Commit
f74255e
1 Parent(s): b794742

reverted back to old app.py

Browse files
Files changed (1) hide show
  1. tortoise/models/utils.py +3 -1
tortoise/models/utils.py CHANGED
@@ -11,7 +11,9 @@ import progressbar
11
 
12
  D_STEM = "https://drive.google.com/uc?id="
13
 
14
- DEFAULT_MODELS_DIR = "/app/tortoise/models/pretrained_models"
 
 
15
 
16
  MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR", DEFAULT_MODELS_DIR)
17
 
 
11
 
12
  D_STEM = "https://drive.google.com/uc?id="
13
 
14
+ DEFAULT_MODELS_DIR = os.path.join(
15
+ os.path.expanduser("~"), ".cache", "tortoise", "models"
16
+ )
17
 
18
  MODELS_DIR = os.environ.get("TORTOISE_MODELS_DIR", DEFAULT_MODELS_DIR)
19