datacipen commited on
Commit
c6947d5
1 Parent(s): 1472131

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -6
main.py CHANGED
@@ -42,8 +42,8 @@ model = HuggingFaceEndpoint(
42
  repo_id=repo_id, max_new_tokens=5000, temperature=1.0, task="text2text-generation", streaming=True
43
  )
44
  os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
45
- embeddings = HuggingFaceEmbeddings()
46
- index_name = "all-venus"
47
  #pc = Pinecone(
48
  # api_key=os.environ['PINECONE_API_KEY']
49
  #)
@@ -53,10 +53,10 @@ os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
53
  #context = ""
54
  #for result in xc['matches']:
55
  # context = context + result['metadata']['text']
56
- vectorstore = PineconeVectorStore(
57
- index_name=index_name, embedding=embeddings
58
- )
59
- retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'categorie': {'$eq': 'bibliographie-OPP-DGDIN'}}})
60
 
61
  @cl.on_chat_start
62
  async def on_chat_start():
 
42
  repo_id=repo_id, max_new_tokens=5000, temperature=1.0, task="text2text-generation", streaming=True
43
  )
44
  os.environ['PINECONE_API_KEY'] = os.environ['PINECONE_API_KEY']
45
+ embeddings = HuggingFaceEmbeddings()
46
+ index_name = "all-venus"
47
  #pc = Pinecone(
48
  # api_key=os.environ['PINECONE_API_KEY']
49
  #)
 
53
  #context = ""
54
  #for result in xc['matches']:
55
  # context = context + result['metadata']['text']
56
+ vectorstore = PineconeVectorStore(
57
+ index_name=index_name, embedding=embeddings
58
+ )
59
+ retriever = vectorstore.as_retriever(search_type="similarity_score_threshold", search_kwargs={"score_threshold": .7, "k": 150,"filter": {'categorie': {'$eq': 'bibliographie-OPP-DGDIN'}}})
60
 
61
  @cl.on_chat_start
62
  async def on_chat_start():