Dhritiman Sagar commited on
Commit
c04d432
·
1 Parent(s): d2009f4

Switch to large model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -43,9 +43,9 @@ Context:
43
  def get_rag_chain():
44
  """Fetches a simple RAG chain"""
45
  prompt = ChatPromptTemplate.from_template(RAG_PROMPT)
46
- embedding = OpenAIEmbeddings(model='text-embedding-3-small')
47
  retriever = QdrantVectorStore.from_existing_collection(
48
- collection_name='ai_ethics_te3_small',
49
  embedding=embedding,
50
  url=os.environ.get('QDRANT_DB'),
51
  api_key=os.environ.get('QDRANT_API_KEY')
 
43
  def get_rag_chain():
44
  """Fetches a simple RAG chain"""
45
  prompt = ChatPromptTemplate.from_template(RAG_PROMPT)
46
+ embedding = OpenAIEmbeddings(model='text-embedding-3-large')
47
  retriever = QdrantVectorStore.from_existing_collection(
48
+ collection_name='ai_ethics_te3_large',
49
  embedding=embedding,
50
  url=os.environ.get('QDRANT_DB'),
51
  api_key=os.environ.get('QDRANT_API_KEY')