AravindKumarRajendran commited on
Commit
cbb31e2
·
1 Parent(s): d67c1ff
Files changed (1) hide show
  1. app.py +1 -1
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('models/classes.txt')
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