Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = "./
|
| 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
|