test

run.py

@@ -12,7 +12,7 @@ def pred(src):
 
     # Create an instance of the model architecture with the loaded configuration
     my_sam_model = SamModel(config=model_config)
-    # #Update the model by loading the weights from saved file.
+    # #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')))
 
     new_image = np.array(Image.open(src).convert("RGB"))