Spaces:
Runtime error
Runtime error
debug: show embedding
Browse files
app.py
CHANGED
@@ -8,7 +8,6 @@ from llama_index import GPTSimpleVectorIndex
|
|
8 |
from openai.embeddings_utils import get_embedding, cosine_similarity
|
9 |
|
10 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
11 |
-
|
12 |
passwd = os.getenv("PASSWD_SECRET")
|
13 |
|
14 |
title = "Confidential forensics tool with ChatGPT"
|
@@ -26,7 +25,8 @@ def search_emails(opt, message, n=3):
|
|
26 |
try:
|
27 |
embedding = get_embedding(message)
|
28 |
message = ""
|
29 |
-
df['
|
|
|
30 |
|
31 |
message_tmp = df.sort_values('similarities', ascending=False).head(n)
|
32 |
message_tmp = [(row.file, row.body, row.similarities) for index, row in message_tmp.iterrows()]
|
|
|
8 |
from openai.embeddings_utils import get_embedding, cosine_similarity
|
9 |
|
10 |
openai.api_key = os.getenv("OPENAI_API_KEY")
|
|
|
11 |
passwd = os.getenv("PASSWD_SECRET")
|
12 |
|
13 |
title = "Confidential forensics tool with ChatGPT"
|
|
|
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)
|
32 |
message_tmp = [(row.file, row.body, row.similarities) for index, row in message_tmp.iterrows()]
|