tamas.kiss commited on
Commit
052aaff
1 Parent(s): e0951b5

bugfix: Removed conversion to list

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ def cached(func):
88
  def create_embedding(text: str):
89
  embed_text = openai.embeddings.create(input=text, model="text-embedding-ada-002").data[0].embedding
90
 
91
- return embed_text.tolist()
92
 
93
  def query_from_pinecone(query, top_k=3):
94
  embedding = create_embedding(query)
 
88
  def create_embedding(text: str):
89
  embed_text = openai.embeddings.create(input=text, model="text-embedding-ada-002").data[0].embedding
90
 
91
+ return embed_text
92
 
93
  def query_from_pinecone(query, top_k=3):
94
  embedding = create_embedding(query)