esra2001 commited on
Commit
4239359
·
verified ·
1 Parent(s): 6267f88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -3
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
- chroma_db_path = "./chroma_db"
 
 
 
 
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="./chroma_db",
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