davidlee1102 commited on
Commit
7b4219a
1 Parent(s): 332bd83

Update model

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -66,4 +66,7 @@ if st.button("Classify"):
66
  distance = euclidean(result[0], result[i])
67
  distance_list.append(1 / (1 + distance))
68
 
69
- col2.write(f"len image: {distance_list}")
 
 
 
 
66
  distance = euclidean(result[0], result[i])
67
  distance_list.append(1 / (1 + distance))
68
 
69
+ max_index = np.argmax(distance_list)
70
+ image_matches = image_list[max_index].get("name")
71
+
72
+ col2.write(f"len image: {image_matches}")