Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ ip_ckpt = hf_hub_download(repo_id="CiaraRowles/stylecodes", filename="stylecodes
|
|
14 |
device = "cuda"
|
15 |
|
16 |
example_path = os.path.join(os.path.dirname(__file__), 'test_images')
|
17 |
-
|
18 |
-
list_path = [os.path.join(example_path,img) for img in
|
19 |
|
20 |
cv2.setNumThreads(1)
|
21 |
|
|
|
14 |
device = "cuda"
|
15 |
|
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 |
|