ShutterStack commited on
Commit
5d10236
·
verified ·
1 Parent(s): ca24d65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ os.makedirs(UPLOAD_FOLDER, exist_ok=True)
19
  os.makedirs(RESULTS_FOLDER, exist_ok=True)
20
 
21
  # Model and data paths - Hugging Face assumes these are in the root directory
22
- MODEL_CLASSIFICATION_PATH = "classification.pt"
23
- MODEL_DETECTION_PATH = "detection.pt"
24
 
25
  # Load YOLOv8 models
26
  try:
 
19
  os.makedirs(RESULTS_FOLDER, exist_ok=True)
20
 
21
  # Model and data paths - Hugging Face assumes these are in the root directory
22
+ MODEL_CLASSIFICATION_PATH = "models/classification.pt"
23
+ MODEL_DETECTION_PATH = "models/detection.pt"
24
 
25
  # Load YOLOv8 models
26
  try: