Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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 = "/
|
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",
|