ConradLax commited on
Commit
da3b513
·
1 Parent(s): 2a578d3

fix: add launcher method

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. onLaunch.py +20 -0
Dockerfile CHANGED
@@ -7,6 +7,8 @@ WORKDIR /code
7
 
8
  COPY ./requirements.txt /code/requirements.txt
9
 
 
 
10
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
11
 
12
  COPY . .
 
7
 
8
  COPY ./requirements.txt /code/requirements.txt
9
 
10
+ RUN python ./onLaunch.py
11
+
12
  RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
13
 
14
  COPY . .
onLaunch.py ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import platform, os
2
+
3
+
4
+ if platform.system() == "Windows":
5
+ print("Windows detected. Assigning cache directory to Transformers in AppData\Local.")
6
+ transformers_cache_directory = os.path.join(os.getenv('LOCALAPPDATA'), 'transformers_cache')
7
+ if not os.path.exists(transformers_cache_directory):
8
+ try:
9
+ os.mkdir(transformers_cache_directory)
10
+ print(f"First launch. Directory '{transformers_cache_directory}' created successfully.")
11
+ except OSError as e:
12
+ print(f"Error creating directory '{transformers_cache_directory}': {e}")
13
+ else:
14
+ print(f"Directory '{transformers_cache_directory}' already exists.")
15
+ os.environ['TRANSFORMERS_CACHE'] = transformers_cache_directory
16
+ print("Environment variable assigned.")
17
+ del transformers_cache_directory
18
+
19
+ else:
20
+ print("Windows not detected. Assignment of Transformers cache directory not necessary.")