ishanprogs commited on
Commit
546586d
·
verified ·
1 Parent(s): fe522f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ CAR_PART_CLASSES = [
33
  NUM_CAR_PART_CLASSES = len(CAR_PART_CLASSES)
34
 
35
  CLIP_TEXT_FEATURES_PATH = "./clip_text_features.pt"
36
- DAMAGE_MODEL_WEIGHTS_PATH = "./model_best.pt"
37
  PART_MODEL_WEIGHTS_PATH = "./partdetection_yolobest.pt"
38
  DEFAULT_DAMAGE_PRED_THRESHOLD = 0.4
39
  DEFAULT_PART_PRED_THRESHOLD = 0.3
 
33
  NUM_CAR_PART_CLASSES = len(CAR_PART_CLASSES)
34
 
35
  CLIP_TEXT_FEATURES_PATH = "./clip_text_features.pt"
36
+ DAMAGE_MODEL_WEIGHTS_PATH = "./best.pt"
37
  PART_MODEL_WEIGHTS_PATH = "./partdetection_yolobest.pt"
38
  DEFAULT_DAMAGE_PRED_THRESHOLD = 0.4
39
  DEFAULT_PART_PRED_THRESHOLD = 0.3