Esmael-Saleh commited on
Commit
00ecd80
1 Parent(s): 9b0587c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,8 +8,9 @@ from fastapi import FastAPI, HTTPException
8
 
9
  import os
10
 
11
- cache_dir = "/path/to/your/specific/cache_directory"
12
- os.makedirs(cache_dir, exist_ok=True) # Create the directory if it doesn't exist
 
13
  os.environ["TRANSFORMERS_CACHE"] = cache_dir
14
 
15
 
 
8
 
9
  import os
10
 
11
+ # Use a directory in your home directory
12
+ cache_dir = os.path.expanduser("~/transformers_cache")
13
+ os.makedirs(cache_dir, exist_ok=True)
14
  os.environ["TRANSFORMERS_CACHE"] = cache_dir
15
 
16