Update app.py

app.py

@@ -143,19 +143,19 @@ def run_model_pipeline(model, original_image, comparison_image, threshold=0.3):
         with col2:
             st.image(img2, caption="Comparison Signature", use_column_width=True)
 
-        # Visualize heatmaps from extracted model features
-        st.subheader("Feature Heatmaps")
-        col3, col4 = st.columns(2)
+        # # Visualize heatmaps from extracted model features
+        # st.subheader("Feature Heatmaps")
+        # col3, col4 = st.columns(2)
         
-        with col3:
-            x1 = rearrange(img1_tensor.float(), 'b h w c -> b c h w')
-            fig1 = visualize_heatmap(model, x1)
-            st.pyplot(fig1)
+        # with col3:
+        #     x1 = rearrange(img1_tensor.float(), 'b h w c -> b c h w')
+        #     fig1 = visualize_heatmap(model, x1)
+        #     st.pyplot(fig1)
             
-        with col4:
-            x2 = rearrange(img2_tensor.float(), 'b h w c -> b c h w')
-            fig2 = visualize_heatmap(model, x2)
-            st.pyplot(fig2)
+        # with col4:
+        #     x2 = rearrange(img2_tensor.float(), 'b h w c -> b c h w')
+        #     fig2 = visualize_heatmap(model, x2)
+        #     st.pyplot(fig2)
 
     else:
         st.write("Please upload both the original and comparison signatures.")