ConradLax commited on
Commit
2166f46
·
1 Parent(s): d26b712

fix: dynamic cache dir

Browse files
Files changed (1) hide show
  1. main.py +22 -1
main.py CHANGED
@@ -1,6 +1,27 @@
1
  import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
- os.environ['HF_HOME'] = './cache'
4
 
5
  from flask import Flask, render_template, request, jsonify
6
 
 
1
  import os
2
+ import platform
3
+
4
+ # os.environ['HF_HOME'] = './cache'
5
+
6
+ if platform.system() == "Windows":
7
+ print("Windows detected. Assigning cache directory to Transformers in AppData\Local.")
8
+ transformers_cache_directory = os.path.join(os.getenv('LOCALAPPDATA'), 'transformers_cache')
9
+ if not os.path.exists(transformers_cache_directory):
10
+ try:
11
+ os.mkdir(transformers_cache_directory)
12
+ print(f"First launch. Directory '{transformers_cache_directory}' created successfully.")
13
+ except OSError as e:
14
+ print(f"Error creating directory '{transformers_cache_directory}': {e}")
15
+ else:
16
+ print(f"Directory '{transformers_cache_directory}' already exists.")
17
+ os.environ['TRANSFORMERS_CACHE'] = transformers_cache_directory
18
+ print("Environment variable assigned.")
19
+ del transformers_cache_directory
20
+
21
+ else:
22
+ print("Windows not detected. Assignment of Transformers cache directory not necessary.")
23
+
24
 
 
25
 
26
  from flask import Flask, render_template, request, jsonify
27