Skip to content

Commit ae74458

Browse files
committed
fix: update aircraft model to use correct weights filename
1 parent 5e1d4ac commit ae74458

File tree

1 file changed

+2
-2
lines changed
  • src/aws/osml/models/aircraft

1 file changed

+2
-2
lines changed

src/aws/osml/models/aircraft/app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2023-2024 Amazon.com, Inc. or its affiliates.
1+
# Copyright 2023-2025 Amazon.com, Inc. or its affiliates.
22

33
import json
44
import os
@@ -50,7 +50,7 @@ def build_predictor() -> DefaultPredictor:
5050
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
5151
# Path to the model weights
5252
cfg.MODEL.WEIGHTS = os.getenv(
53-
os.path.join("MODEL_WEIGHTS"), os.path.join("/home/osml-models/assets/", "model_weights.pth")
53+
os.path.join("MODEL_WEIGHTS"), os.path.join("/home/osml-models/assets/", "aircraft_model_weights.pth")
5454
)
5555

5656
# Build the detectron2 default predictor

0 commit comments

Comments
 (0)