update

run.py

@@ -3,7 +3,6 @@ import torch
 import numpy as np
 import app
 import os
-from patchify import patchify
 from PIL import Image
 
 
@@ -32,7 +31,7 @@ def pred(src):
     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')))
-
+    print(src)
     new_image = np.array(Image.open(src))
     inputs = processor(new_image, return_tensors="pt")
     my_sam_model.eval()