johnnyfivefingers commited on
Commit
31f4e07
·
1 Parent(s): b646897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
 
3
  from sklearn.metrics.pairwise import cosine_similarity
 
4
 
5
  def greet(array_of_vectors):
6
  # [[0 1],[0 5]]
@@ -18,7 +19,8 @@ def greet(array_of_vectors):
18
  int_arr.append(int(a))
19
  cleaned.append(int_arr)
20
 
21
- similarity_matrix = cosine_similarity(cleaned, cleaned)
 
22
  results = ''
23
  for matrix in similarity_matrix:
24
  results = results + str(matrix) + "\n"
 
1
  import gradio as gr
2
 
3
  from sklearn.metrics.pairwise import cosine_similarity
4
+ from sklearn.metrics.pairwise import euclidean_distances
5
 
6
  def greet(array_of_vectors):
7
  # [[0 1],[0 5]]
 
19
  int_arr.append(int(a))
20
  cleaned.append(int_arr)
21
 
22
+ # similarity_matrix = cosine_similarity(cleaned, cleaned)
23
+ similarity_matrix = euclidean_distances(cleaned, cleaned)
24
  results = ''
25
  for matrix in similarity_matrix:
26
  results = results + str(matrix) + "\n"