Update app.py

app.py

@@ -232,6 +232,7 @@ def prod_function(app, prod_path, webcam_path):
         face_embedding = torch.from_numpy(np.array(face_embedding))
         
         similarity_score.append(F.cosine_similarity(face_embedding,webcam_emb, dim=0))
+        similarity_score = torch.from_numpy(np.array(similarity_score))
         #distance = F.pairwise_distance(emb, emb_prod)
         #prod_preds.append(distance)
     
@@ -285,7 +286,7 @@ with app_tab:
         with st.spinner("Wait for it...", show_time=True):
             # Run the predictions
             prediction = prod_function(app, image_paths, picture)
-            prediction = torch.cat(prediction, 0).to(device)
+            #prediction = torch.cat(prediction, 0).to(device)
             match_idx = torch.argmax(prediction)
             st.write(prediction)
             st.write(image_paths)