MNCJihun commited on
Commit
2a50db8
·
1 Parent(s): 98d3545

add sample images

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