Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -121,7 +121,7 @@ def run_model_pipeline(model, original_image, comparison_image, threshold=0.5):
|
|
121 |
st.success('images loaded')
|
122 |
|
123 |
# Get model embeddings/probabilites
|
124 |
-
output1, output2 = model(img1_tensor, img2_tensor)
|
125 |
st.success('outputs extracted')
|
126 |
|
127 |
# Compute similarity
|
|
|
121 |
st.success('images loaded')
|
122 |
|
123 |
# Get model embeddings/probabilites
|
124 |
+
output1, output2 = model(img1_tensor.float(), img2_tensor.float())
|
125 |
st.success('outputs extracted')
|
126 |
|
127 |
# Compute similarity
|