Chaerin5 commited on
Commit
ec4b2bc
·
1 Parent(s): 644d803
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ LENGTH = 480
31
  placeholder = cv2.cvtColor(cv2.imread("placeholder.png"), cv2.COLOR_BGR2RGB)
32
  NEW_MODEL = True
33
  MODEL_EPOCH = 6
34
- HF = False
35
  pre_device = "cpu" if HF else "cuda"
36
  spaces_60_fn = spaces.GPU(duration=60) if HF else (lambda f: f)
37
  spaces_120_fn = spaces.GPU(duration=60) if HF else (lambda f: f)
 
31
  placeholder = cv2.cvtColor(cv2.imread("placeholder.png"), cv2.COLOR_BGR2RGB)
32
  NEW_MODEL = True
33
  MODEL_EPOCH = 6
34
+ HF = True
35
  pre_device = "cpu" if HF else "cuda"
36
  spaces_60_fn = spaces.GPU(duration=60) if HF else (lambda f: f)
37
  spaces_120_fn = spaces.GPU(duration=60) if HF else (lambda f: f)