Update sam/utils.py

sam/utils.py

@@ -230,10 +230,15 @@ def visualize_attention(arr, filename):
 
 
 # Build config
-def build_sam_config(config_path):
-  with open(config_path, 'r') as infile:
-    config = json.load(infile)
-
-  sam_checkpoint = config['model']['sam_checkpoint']
-  model_type = config['model']['model_type']
-  return sam_checkpoint, model_type
+# def build_sam_config(config_path):
+#   with open(config_path, 'r') as infile:
+#     config = json.load(infile)
+
+#   sam_checkpoint = config['model']['sam_checkpoint']
+#   model_type = config['model']['model_type']
+#   return sam_checkpoint, model_type
+
+def build_sam_config(config):
+    sam_checkpoint = config.sam.sam_checkpoint
+    model_type = config.sam.model_type
+    return sam_checkpoint, model_type