Spaces:
Runtime error
Runtime error
add sample images
Browse files
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)
|