Spaces:
Runtime error
Runtime error
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 |
|