CiaraRowles commited on
Commit
f0c8de8
·
verified ·
1 Parent(s): ef9bd00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- list = os.listdir(example_path)
18
- list_path = [os.path.join(example_path,img) for img in list]
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