Spaces:
Running
on
Zero
Running
on
Zero
hf
Browse files
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 =
|
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)
|