r3gm commited on
Commit
9b20e87
·
verified ·
1 Parent(s): e9bbfb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -36,6 +36,11 @@ import aoti
36
 
37
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
38
  warnings.filterwarnings("ignore")
 
 
 
 
 
39
 
40
  # --- FRAME EXTRACTION JS & LOGIC ---
41
 
 
36
 
37
  os.environ["TOKENIZERS_PARALLELISM"] = "true"
38
  warnings.filterwarnings("ignore")
39
+ IS_ZERO_GPU = bool(os.getenv("SPACES_ZERO_GPU"))
40
+
41
+ if IS_ZERO_GPU:
42
+ print("Loading...")
43
+ subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
44
 
45
  # --- FRAME EXTRACTION JS & LOGIC ---
46