Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,8 @@ title = "Confidential forensics tool with ChatGPT"
|
|
| 14 |
examples = ["Who is Phillip Allen?", "What the project in Austin is about?", "Give me more details about the real estate project"]
|
| 15 |
|
| 16 |
index = GPTSimpleVectorIndex.load_from_disk('email.json')
|
| 17 |
-
|
|
|
|
| 18 |
df2 = df["subject"].replace([None], "<no title>").head(10).to_frame()
|
| 19 |
|
| 20 |
def search_emails(opt, message, n=3):
|
|
@@ -25,8 +26,6 @@ def search_emails(opt, message, n=3):
|
|
| 25 |
try:
|
| 26 |
embedding = get_embedding(message)
|
| 27 |
message = ""
|
| 28 |
-
df['embedding'] = df['embedding'].str.decode('utf-8').astype(float)
|
| 29 |
-
|
| 30 |
df['similarities'] = df.embedding.apply(func=(lambda x: cosine_similarity(x, embedding)))
|
| 31 |
|
| 32 |
message_tmp = df.sort_values('similarities', ascending=False).head(n)
|
|
@@ -131,4 +130,5 @@ with gr.Blocks(title=title) as demo:
|
|
| 131 |
btn_send2.click(search_emails, [opt, msg2], [opt, msg2])
|
| 132 |
|
| 133 |
if __name__ == "__main__":
|
| 134 |
-
demo.launch(auth=("mithril", passwd))
|
|
|
|
|
|
| 14 |
examples = ["Who is Phillip Allen?", "What the project in Austin is about?", "Give me more details about the real estate project"]
|
| 15 |
|
| 16 |
index = GPTSimpleVectorIndex.load_from_disk('email.json')
|
| 17 |
+
|
| 18 |
+
df = pd.read_feather("emails.feather")
|
| 19 |
df2 = df["subject"].replace([None], "<no title>").head(10).to_frame()
|
| 20 |
|
| 21 |
def search_emails(opt, message, n=3):
|
|
|
|
| 26 |
try:
|
| 27 |
embedding = get_embedding(message)
|
| 28 |
message = ""
|
|
|
|
|
|
|
| 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)
|
|
|
|
| 130 |
btn_send2.click(search_emails, [opt, msg2], [opt, msg2])
|
| 131 |
|
| 132 |
if __name__ == "__main__":
|
| 133 |
+
# demo.launch(auth=("mithril", passwd))
|
| 134 |
+
demo.launch()
|