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

add sample images

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -21,9 +21,9 @@ 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(), img_samples[0])))
25
- assert False
26
- img_samples = [os.path.join(os.getcwd(), './samples/', img) for img in img_samples]
27
  MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
28
  MODEL_PATH = "/tmp/best_model.pth"
29
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
 
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"
28
  MODEL_PATH = "/tmp/best_model.pth"
29
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)