hoangkha1810 commited on
Commit
c4b2683
·
verified ·
1 Parent(s): 50adcef

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -2
main.py CHANGED
@@ -17,10 +17,14 @@ logger = logging.getLogger(__name__)
17
 
18
  app = FastAPI()
19
 
20
- # Tạo thư mục lưu video
21
- output_dir = "/app/outputs"
22
  os.makedirs(output_dir, exist_ok=True)
23
 
 
 
 
 
24
  # Constants
25
  bases = {
26
  "Cartoon": "frankjoshua/toonyou_beta6",
 
17
 
18
  app = FastAPI()
19
 
20
+ # Tạo thư mục lưu video trong /tmp (có quyền ghi)
21
+ output_dir = "/tmp/outputs"
22
  os.makedirs(output_dir, exist_ok=True)
23
 
24
+ # Thiết lập thư mục cache cho Hugging Face
25
+ os.environ["HF_HOME"] = "/tmp/huggingface_cache"
26
+ os.makedirs(os.environ["HF_HOME"], exist_ok=True)
27
+
28
  # Constants
29
  bases = {
30
  "Cartoon": "frankjoshua/toonyou_beta6",