Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,11 @@ os.environ["HF_HOME"] = "/tmp/huggingface_cache"
|
|
4 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
|
5 |
cache_dir = os.environ["HF_HOME"]
|
6 |
|
|
|
|
|
|
|
|
|
|
|
7 |
import cv2
|
8 |
import numpy as np
|
9 |
import torch
|
|
|
4 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
|
5 |
cache_dir = os.environ["HF_HOME"]
|
6 |
|
7 |
+
|
8 |
+
if not os.path.exists(cache_dir):
|
9 |
+
os.makedirs(cache_dir, exist_ok=True)
|
10 |
+
|
11 |
+
|
12 |
import cv2
|
13 |
import numpy as np
|
14 |
import torch
|