We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5e1d4ac commit ae74458Copy full SHA for ae74458
src/aws/osml/models/aircraft/app.py
@@ -1,4 +1,4 @@
1
-# Copyright 2023-2024 Amazon.com, Inc. or its affiliates.
+# Copyright 2023-2025 Amazon.com, Inc. or its affiliates.
2
3
import json
4
import os
@@ -50,7 +50,7 @@ def build_predictor() -> DefaultPredictor:
50
cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
51
# Path to the model weights
52
cfg.MODEL.WEIGHTS = os.getenv(
53
- os.path.join("MODEL_WEIGHTS"), os.path.join("/home/osml-models/assets/", "model_weights.pth")
+ os.path.join("MODEL_WEIGHTS"), os.path.join("/home/osml-models/assets/", "aircraft_model_weights.pth")
54
)
55
56
# Build the detectron2 default predictor
0 commit comments