Update app.py

app.py

@@ -109,7 +109,8 @@ def run_model_pipeline(model, original_image, comparison_image, threshold=0.5):
         # convert to tensors and add batch dimensions to match model input shape
         img1_tensor = torch.unsqueeze(torch.as_tensor(img1), 0)
         img2_tensor = torch.unsqueeze(torch.as_tensor(img2), 0)
-
+        st.success('images loaded')
+        
         # Get model embeddings/probabilites
         output1, output2 = model(img1_tensor, img2_tensor)
         st.success('outputs extracted')
@@ -147,9 +148,9 @@ def run_model_pipeline(model, original_image, comparison_image, threshold=0.5):
         st.write("Please upload both the original and comparison signatures.")
 
 # Run the model pipeline if a button is clicked
-if st.button("Run Model Pipeline"):
+if st.button("Run Model Pipeline for images"):
     model = load_pipeline()
     
 # button click to process images
-if st.button("Process Images"):
+# if st.button("Process Images"):
     run_model_pipeline(model, original_image, comparison_image)