Spaces:
Running
on
Zero
Running
on
Zero
Update stf/stf-api-alternative/src/stf_alternative/preprocess.py
Browse files
stf/stf-api-alternative/src/stf_alternative/preprocess.py
CHANGED
|
@@ -2,7 +2,7 @@ import gc
|
|
| 2 |
import os
|
| 3 |
from pathlib import Path
|
| 4 |
|
| 5 |
-
import cv2
|
| 6 |
from PIL import Image
|
| 7 |
from tqdm import tqdm
|
| 8 |
|
|
|
|
| 2 |
import os
|
| 3 |
from pathlib import Path
|
| 4 |
|
| 5 |
+
import cv2; cv2.setNumThreads(0); cv2.ocl.setUseOpenCL(False)
|
| 6 |
from PIL import Image
|
| 7 |
from tqdm import tqdm
|
| 8 |
|