cloud-sean commited on
Commit
48879d4
·
1 Parent(s): a8310bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -48,7 +48,10 @@ def upload_pdf(file, pdf_text, embeddings, vectorstore, azure_embeddings, qa, pr
48
  embeddings.append(emb)
49
 
50
 
51
- azure_embeddings = OpenAIEmbeddings(document_model_name="text-embedding-ada-002",query_model_name="text-embedding-ada-002")
 
 
 
52
  vectorstore = Chroma("collection", embedding_function=azure_embeddings)
53
 
54
  vectorstore._collection.add(
 
48
  embeddings.append(emb)
49
 
50
 
51
+ azure_embeddings = OpenAIEmbeddings(
52
+ deployment="text-embedding-ada-002",
53
+ model="ytext-embedding-ada-002",
54
+ )
55
  vectorstore = Chroma("collection", embedding_function=azure_embeddings)
56
 
57
  vectorstore._collection.add(