moved model to see if itll fix the bug
Browse files- handler.py +1 -1
- tf_model.h5 → models/tf_model.h5 +0 -0
handler.py
CHANGED
@@ -9,7 +9,7 @@ class EndpointHandler():
|
|
9 |
# Preload all the elements you are going to need at inference.
|
10 |
model_type = "vit_b"
|
11 |
# prefix = "/opt/ml/model"
|
12 |
-
model_path = "tf_model.h5"
|
13 |
# model_checkpoint_path = os.path.join(prefix, "sam_vit_h_4b8939.pth")
|
14 |
sam = sam_model_registry[model_type](checkpoint=model_path)
|
15 |
self.predictor = SamPredictor(sam)
|
|
|
9 |
# Preload all the elements you are going to need at inference.
|
10 |
model_type = "vit_b"
|
11 |
# prefix = "/opt/ml/model"
|
12 |
+
model_path = "models/tf_model.h5"
|
13 |
# model_checkpoint_path = os.path.join(prefix, "sam_vit_h_4b8939.pth")
|
14 |
sam = sam_model_registry[model_type](checkpoint=model_path)
|
15 |
self.predictor = SamPredictor(sam)
|
tf_model.h5 → models/tf_model.h5
RENAMED
File without changes
|