tensorkelechi commited on
Commit
1d8a6e3
·
verified ·
1 Parent(s): 5e3554b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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