mayzinoo commited on
Commit
519b85c
·
verified ·
1 Parent(s): 683be4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,13 +3,13 @@ import re
3
  import gradio as gr
4
  from langchain.prompts import PromptTemplate
5
  from langchain_openai import ChatOpenAI
6
- from langchain.vectorstores import Chroma
7
  from langchain.embeddings import HuggingFaceEmbeddings
8
 
9
  # Load embedding model and vector store from persisted DB
10
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
11
  vector_store = Chroma(
12
- embedding=embedding_model,
13
  persist_directory="geometry_db", # relative folder inside your Hugging Face Space
14
  collection_name="geometry_sol"
15
  )
 
3
  import gradio as gr
4
  from langchain.prompts import PromptTemplate
5
  from langchain_openai import ChatOpenAI
6
+ from langchain_chroma import Chroma
7
  from langchain.embeddings import HuggingFaceEmbeddings
8
 
9
  # Load embedding model and vector store from persisted DB
10
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
11
  vector_store = Chroma(
12
+ embedding_function=embedding_model,
13
  persist_directory="geometry_db", # relative folder inside your Hugging Face Space
14
  collection_name="geometry_sol"
15
  )