add config

run.py

@@ -8,11 +8,11 @@ import os
 def pred(src):
     # os.environ['HUGGINGFACE_HUB_HOME'] = './.cache'
     # Load the model configuration
-    # model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
-    # processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
+    model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
+    processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 
     # Create an instance of the model architecture with the loaded configuration
-    my_sam_model = SamModel()
+    my_sam_model = SamModel(config=model_config)
     #Update the model by loading the weights from saved file.
     my_sam_model.load_state_dict(torch.load("sam_model.pth", map_location=torch.device('cpu')))