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