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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -8,7 +8,10 @@ from fastapi import FastAPI, HTTPException
8
 
9
  import os
10
 
11
- os.environ["TRANSFORMERS_CACHE"] = "/transformers_cache/cache/"
 
 
 
12
 
13
 
14
  # Initialize FastAPI app
 
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
 
16
 
17
  # Initialize FastAPI app