Update main.py

main.py

@@ -1,4 +1,4 @@
-from langchain_community.document_loaders import TextLoader
+from langchain_community.document_loaders import TextLoader , PyPDFLoader
 from langchain_community.embeddings import HuggingFaceBgeEmbeddings
 from langchain_community.vectorstores import FAISS
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -7,7 +7,9 @@ from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from langchain.chains import RetrievalQA
 import torch
-loader = TextLoader("info.txt")
+
+loader = PyPDFLoader('bipolar.pdf')
+# loader = TextLoader("info.txt")
 docs = loader.load()
 text_splitter = RecursiveCharacterTextSplitter()
 # text_splitter = RecursiveCharacterTextSplitter(chunk_size=100, chunk_overlap=20)