Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,8 @@ from langchain.prompts import PromptTemplate
|
|
36 |
from langchain.schema.runnable import RunnablePassthrough
|
37 |
from sentence_transformers import CrossEncoder
|
38 |
from langchain_openai import ChatOpenAI
|
|
|
|
|
39 |
|
40 |
load_dotenv()
|
41 |
|
@@ -55,14 +57,18 @@ def detect_intent(text):
|
|
55 |
result = classifier(text, candidate_labels=["question", "greeting", "small talk", "feedback", "thanks"])
|
56 |
label = result["labels"][0]
|
57 |
return label.lower()
|
58 |
-
|
59 |
-
|
|
|
|
|
|
|
|
|
60 |
chroma_client = chromadb.PersistentClient(path=chroma_db_path)
|
61 |
|
62 |
data = chroma_client.get_collection(name="my_dataaaa")
|
63 |
vectorstore = Chroma(
|
64 |
collection_name="my_dataaaa",
|
65 |
-
persist_directory="./
|
66 |
embedding_function=embeddings_model
|
67 |
)
|
68 |
|
|
|
36 |
from langchain.schema.runnable import RunnablePassthrough
|
37 |
from sentence_transformers import CrossEncoder
|
38 |
from langchain_openai import ChatOpenAI
|
39 |
+
import zipfile
|
40 |
+
|
41 |
|
42 |
load_dotenv()
|
43 |
|
|
|
57 |
result = classifier(text, candidate_labels=["question", "greeting", "small talk", "feedback", "thanks"])
|
58 |
label = result["labels"][0]
|
59 |
return label.lower()
|
60 |
+
|
61 |
+
if not os.path.exists("./chroma_db_copy"):
|
62 |
+
with zipfile.ZipFile("chroma_db_copy.zip", "r") as zip_ref:
|
63 |
+
zip_ref.extractall("./")
|
64 |
+
|
65 |
+
chroma_db_path = "./chroma_db_copy"
|
66 |
chroma_client = chromadb.PersistentClient(path=chroma_db_path)
|
67 |
|
68 |
data = chroma_client.get_collection(name="my_dataaaa")
|
69 |
vectorstore = Chroma(
|
70 |
collection_name="my_dataaaa",
|
71 |
+
persist_directory="./chroma_db_copy",
|
72 |
embedding_function=embeddings_model
|
73 |
)
|
74 |
|