Commit
·
cbb31e2
1
Parent(s):
d67c1ff
path fix
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
|
|
14 |
|
15 |
# Path configurations
|
16 |
MODEL_PATH = Path('src/model_10.pth')
|
17 |
-
CLASSES_PATH = Path('
|
18 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
19 |
|
20 |
# Image preprocessing - using the same transforms as training
|
|
|
14 |
|
15 |
# Path configurations
|
16 |
MODEL_PATH = Path('src/model_10.pth')
|
17 |
+
CLASSES_PATH = Path('src/classes.txt')
|
18 |
DEVICE = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
|
19 |
|
20 |
# Image preprocessing - using the same transforms as training
|