malavikapradeep2001 commited on
Commit
af59754
·
unverified ·
1 Parent(s): f20a61d

Update backend/app.py

Browse files
Files changed (1) hide show
  1. backend/app.py +3 -0
backend/app.py CHANGED
@@ -5,6 +5,9 @@ os.environ["TORCH_HOME"] = "/tmp/torch"
5
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
6
  os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
7
 
 
 
 
8
  from huggingface_hub import login
9
 
10
 
 
5
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
6
  os.environ["YOLO_CONFIG_DIR"] = "/tmp/Ultralytics"
7
 
8
+ for d in ["/tmp/huggingface", "/tmp/Ultralytics", "/tmp/matplotlib", "/tmp/torch", "/root/.cache"]:
9
+ shutil.rmtree(d, ignore_errors=True)
10
+
11
  from huggingface_hub import login
12
 
13