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")
|