MNCJihun commited on
Commit
a7c1b55
·
1 Parent(s): 4fce033

add sample images

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,8 +17,9 @@ test_transforms = A.Compose(
17
  ToTensorV2(),
18
  ]
19
  )
20
-
21
  img_samples = os.listdir('./sample/')
 
22
  MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
23
  MODEL_PATH = "/tmp/best_model.pth"
24
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
 
17
  ToTensorV2(),
18
  ]
19
  )
20
+ print(os.getcwd())
21
  img_samples = os.listdir('./sample/')
22
+ img_samples = [os.path.join('./samples/', img) for im in img_samples]
23
  MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
24
  MODEL_PATH = "/tmp/best_model.pth"
25
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)