Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ device = "cuda"
|
|
16 |
example_path = os.path.join(os.path.dirname(__file__), 'test_images')
|
17 |
list_dir = os.listdir(example_path)
|
18 |
list_path = [os.path.join(example_path,img) for img in list_dir]
|
19 |
-
|
20 |
cv2.setNumThreads(1)
|
21 |
|
22 |
@spaces.GPU(enable_queue=True)
|
|
|
16 |
example_path = os.path.join(os.path.dirname(__file__), 'test_images')
|
17 |
list_dir = os.listdir(example_path)
|
18 |
list_path = [os.path.join(example_path,img) for img in list_dir]
|
19 |
+
print("all paths ", list_path)
|
20 |
cv2.setNumThreads(1)
|
21 |
|
22 |
@spaces.GPU(enable_queue=True)
|