Spaces:
Runtime error
Runtime error
debug: show embedding
Browse files- app.py +1 -1
- 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 |
-
|
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
|