aradootle commited on
Commit
1810c2f
1 Parent(s): 7f7a4fc

moved model to see if itll fix the bug

Browse files
Files changed (2) hide show
  1. handler.py +1 -1
  2. 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