JoFrost commited on
Commit
da5b0b4
1 Parent(s): a0d26d2

debug: show embedding

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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['similarities'] = df.embedding.apply(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()]
 
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()]