Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ inference = torch.load('fine_tune_resnet.pth', map_location=torch.device('cpu'))
|
|
15 |
inference.eval()
|
16 |
|
17 |
|
18 |
-
example = [
|
19 |
|
20 |
def classifier(image):
|
21 |
test_transform = Compose([
|
|
|
15 |
inference.eval()
|
16 |
|
17 |
|
18 |
+
example = [str(i) for i in Path('examples').glob('*')]
|
19 |
|
20 |
def classifier(image):
|
21 |
test_transform = Compose([
|