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

add sample images

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -17,9 +17,10 @@ test_transforms = A.Compose(
17
  ToTensorV2(),
18
  ]
19
  )
 
20
  print(os.getcwd())
21
  img_samples = os.listdir('./sample/')
22
- img_samples = [os.path.join('./samples/', img) for img 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)
 
17
  ToTensorV2(),
18
  ]
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"
25
  MODEL_PATH = "/tmp/best_model.pth"
26
  urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)