.

pages/3_Verify.py

@@ -78,6 +78,7 @@ if st.session_state.predict:
         weights = torch.load("pages/siamese.pth", map_location=torch.device('cpu'))["model_state_dict"]
         model = SiameseNetwork(backbone)
         model.load_state_dict(weights)
+        model.eval()
         img1, img2 = transform_valid(img1).unsqueeze(0), transform_valid(img2).unsqueeze(0)
         prediction = model(img1,img2)
         st.write(f"The signatures match with a probability of {str(round(float(prediction)*100, 2))}%")