Spaces:
Runtime error
Runtime error
add sample images
Browse files
app.py
CHANGED
@@ -19,6 +19,7 @@ test_transforms = A.Compose(
|
|
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"
|
|
|
19 |
)
|
20 |
|
21 |
print(os.getcwd())
|
22 |
+
print(os.listdir(os.getcwd()))
|
23 |
img_samples = os.listdir('./sample/')
|
24 |
img_samples = [os.path.join(os.getcwd(), './samples/', img) for img in img_samples]
|
25 |
MODEL_URL = "https://huggingface.co/caisarl76/HI_motorcycle_trunk_cls_model/resolve/main/best_model.pth"
|