MNCJihun commited on
Commit
cf991f8
·
1 Parent(s): 7041714

add sample images

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -19,6 +19,7 @@ test_transforms = A.Compose(
19
  )
20
 
21
  print(os.getcwd())
 
22
  img_samples = os.listdir('./sample/')
23
  img_samples = [os.path.join(os.getcwd(), './samples/', img) for img in img_samples]
24
  MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
 
19
  )
20
 
21
  print(os.getcwd())
22
+ print(os.listdir(os.getcwd()))
23
  img_samples = os.listdir('./sample/')
24
  img_samples = [os.path.join(os.getcwd(), './samples/', img) for img in img_samples]
25
  MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"