rishab1090 commited on
Commit
b629612
·
verified ·
1 Parent(s): 8c42018

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from pathlib import Path
5
  from ultralytics import YOLO
6
 
7
  # ------------ CONFIG ------------
8
- MODEL_PATH = Path(__file__).parent / "best_rockfall_model.pt"
9
  if not MODEL_PATH.exists():
10
  raise FileNotFoundError(f"Model not found at {MODEL_PATH}. Place your .pt next to app.py or change MODEL_PATH.")
11
 
 
5
  from ultralytics import YOLO
6
 
7
  # ------------ CONFIG ------------
8
+ MODEL_PATH = Path(__file__).parent / "best.pt"
9
  if not MODEL_PATH.exists():
10
  raise FileNotFoundError(f"Model not found at {MODEL_PATH}. Place your .pt next to app.py or change MODEL_PATH.")
11