Spaces:
Sleeping
Sleeping
tensorkelechi
commited on
Commit
•
0d0f527
1
Parent(s):
555682c
Update app.py
Browse files
app.py
CHANGED
@@ -143,19 +143,19 @@ def run_model_pipeline(model, original_image, comparison_image, threshold=0.3):
|
|
143 |
with col2:
|
144 |
st.image(img2, caption="Comparison Signature", use_column_width=True)
|
145 |
|
146 |
-
# Visualize heatmaps from extracted model features
|
147 |
-
st.subheader("Feature Heatmaps")
|
148 |
-
col3, col4 = st.columns(2)
|
149 |
|
150 |
-
with col3:
|
151 |
-
|
152 |
-
|
153 |
-
|
154 |
|
155 |
-
with col4:
|
156 |
-
|
157 |
-
|
158 |
-
|
159 |
|
160 |
else:
|
161 |
st.write("Please upload both the original and comparison signatures.")
|
|
|
143 |
with col2:
|
144 |
st.image(img2, caption="Comparison Signature", use_column_width=True)
|
145 |
|
146 |
+
# # Visualize heatmaps from extracted model features
|
147 |
+
# st.subheader("Feature Heatmaps")
|
148 |
+
# col3, col4 = st.columns(2)
|
149 |
|
150 |
+
# with col3:
|
151 |
+
# x1 = rearrange(img1_tensor.float(), 'b h w c -> b c h w')
|
152 |
+
# fig1 = visualize_heatmap(model, x1)
|
153 |
+
# st.pyplot(fig1)
|
154 |
|
155 |
+
# with col4:
|
156 |
+
# x2 = rearrange(img2_tensor.float(), 'b h w c -> b c h w')
|
157 |
+
# fig2 = visualize_heatmap(model, x2)
|
158 |
+
# st.pyplot(fig2)
|
159 |
|
160 |
else:
|
161 |
st.write("Please upload both the original and comparison signatures.")
|