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
|