Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,20 @@ import numpy as np
|
|
7 |
import faiss
|
8 |
import os
|
9 |
import pickle
|
10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
# === CONFIG ===
|
13 |
openai.api_key = os.environ.get("OPENAI_API_KEY")
|
|
|
7 |
import faiss
|
8 |
import os
|
9 |
import pickle
|
10 |
+
import openai
|
11 |
+
import numpy as np
|
12 |
+
|
13 |
+
def get_embedding(text, model="text-embedding-3-small"):
|
14 |
+
response = openai.embeddings.create(
|
15 |
+
input=[text],
|
16 |
+
model=model
|
17 |
+
)
|
18 |
+
return response.data[0].embedding
|
19 |
+
|
20 |
+
def cosine_similarity(vec1, vec2):
|
21 |
+
vec1 = np.array(vec1)
|
22 |
+
vec2 = np.array(vec2)
|
23 |
+
return np.dot(vec1, vec2) / (np.linalg.norm(vec1) * np.linalg.norm(vec2))
|
24 |
|
25 |
# === CONFIG ===
|
26 |
openai.api_key = os.environ.get("OPENAI_API_KEY")
|