pratikshahp commited on
Commit
5b5cbab
·
verified ·
1 Parent(s): 83bbe50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-ada-002"
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):
 
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):