Update app.py

app.py

@@ -3,21 +3,18 @@ from langchain.llms import Replicate
 from langchain.vectorstores import Pinecone
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.document_loaders import PyPDFLoader
+from langchain.llms import HuggingFaceHub
+from langchain.vectorstores import Chroma
 from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.chains import ConversationalRetrievalChain
 from datasets import load_dataset
 import os
-import pinecone
 
 
 key = os.environ.get('API')
-yeh = os.environ.get('pineapi')
 os.environ["REPLICATE_API_TOKEN"] = key
-pinecone.init(api_key=yeh, environment='gcp-starter')
-
 
 import sentence_transformers
-import faiss
 
 def loading_pdf():
     return "Loading..."
@@ -31,9 +28,8 @@ def pdf_changes(pdf_doc):
     
     embeddings = HuggingFaceEmbeddings()
 
-    index_name = "chatbot"
-    index = pinecone.Index(index_name)
-    vectordb = Pinecone.from_documents(texts, embeddings, index_name=index_name)
+    db = Chroma.from_documents(texts, embeddings)
+    retriever = db.as_retriever(search_kwargs={'k': 2})
 
     llm = Replicate(
         model="a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5",
@@ -42,7 +38,7 @@ def pdf_changes(pdf_doc):
     global qa 
     qa = ConversationalRetrievalChain.from_llm(
         llm,
-        vectordb.as_retriever(search_kwargs={'k': 2}),
+        retriever,
         return_source_documents=True
     )
     return "Ready"