.

pages/3_Signify.py

@@ -70,8 +70,8 @@ st.session_state.predict = st.button('Detect Signature')
 if st.session_state.predict:
     device = torch.device('cpu')
     backbone = models.efficientnet_v2_s(pretrained=True)
-    model = SiameseNetwork(backbone)
     weights = torch.load("pages/siamese.pth", map_location=torch.device('cpu'))["model_state_dict"]
+    model = SiameseNetwork(backbone)
     model.load_state_dict(weights)
     img1, img2 = transform_valid(img1).unsqueeze(0), transform_valid(img2).unsqueeze(0)
     prediction = model(img1,img2)

signify/yolo/detect.py

@@ -22,7 +22,7 @@ from .utils.torch_utils import (load_classifier, select_device,
 
 def detect(image_path):
     opt = {
-    'weights': '../best.pt',
+    'weights': 'best.pt',
     'source': image_path,
     'img_size': 640,
     'conf_thres': 0.25,

signify/yolo/models/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8940b5b18fbda2e984f972f621c9e52c906a4b499533b13ee51f86f53c0d9838
+size 175088335