ovi054 commited on
Commit
38750c0
·
verified ·
1 Parent(s): 8f05d5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
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