Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,9 +35,9 @@ def split_text(text, chunk_size=500, overlap=50):
|
|
| 35 |
def get_embeddings(text):
|
| 36 |
response = openai.Embedding.create(
|
| 37 |
input=text,
|
| 38 |
-
model="text-embedding-
|
| 39 |
)
|
| 40 |
-
return np.array(response
|
| 41 |
|
| 42 |
# Function to retrieve relevant chunks
|
| 43 |
def retrieve_relevant_chunks(query, chunks, chunk_embeddings):
|
|
|
|
| 35 |
def get_embeddings(text):
|
| 36 |
response = openai.Embedding.create(
|
| 37 |
input=text,
|
| 38 |
+
model="text-embedding-3-snmall"
|
| 39 |
)
|
| 40 |
+
return np.array(response.data[0].embedding)
|
| 41 |
|
| 42 |
# Function to retrieve relevant chunks
|
| 43 |
def retrieve_relevant_chunks(query, chunks, chunk_embeddings):
|