JoFrost commited on
Commit
16b9532
·
1 Parent(s): da5b0b4

debug: show embedding

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def search_emails(opt, message, n=3):
25
  try:
26
  embedding = get_embedding(message)
27
  message = ""
28
- print(df['embedding'])
29
  df['similarities'] = df.embedding.apply(func=(lambda x: cosine_similarity(x, embedding)))
30
 
31
  message_tmp = df.sort_values('similarities', ascending=False).head(n)
 
25
  try:
26
  embedding = get_embedding(message)
27
  message = ""
28
+ df['embedding'] = df['embedding'].str.decode('utf-8').astype(float)
29
  df['similarities'] = df.embedding.apply(func=(lambda x: cosine_similarity(x, embedding)))
30
 
31
  message_tmp = df.sort_values('similarities', ascending=False).head(n)
requirements.txt CHANGED
@@ -5,4 +5,5 @@ numpy==1.24.3
5
  openai==0.27.4
6
  plotly==5.14.1
7
  scikit-learn==1.2.2
8
- fastparquet==2023.4.0
 
 
5
  openai==0.27.4
6
  plotly==5.14.1
7
  scikit-learn==1.2.2
8
+ fastparquet==2023.4.0
9
+ pyarrow