sudhir2016 commited on
Commit
1877ee5
·
1 Parent(s): 36f3f56

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -3,11 +3,14 @@ from langchain.vectorstores import Chroma
3
  from langchain.docstore.document import Document
4
  from langchain.embeddings import HuggingFaceEmbeddings
5
  from langchain.text_splitter import CharacterTextSplitter
 
6
  embeddings = HuggingFaceEmbeddings()
7
- g=open('Gita.txt')
8
- Gita=g.read()
 
 
9
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
10
- texts = text_splitter.split_text(Gita)
11
  docsearch = Chroma.from_texts(texts, embeddings)
12
  def answer(query):
13
  docs = docsearch.similarity_search(query)
 
3
  from langchain.docstore.document import Document
4
  from langchain.embeddings import HuggingFaceEmbeddings
5
  from langchain.text_splitter import CharacterTextSplitter
6
+ from langchain.document_loaders import TextLoader
7
  embeddings = HuggingFaceEmbeddings()
8
+ #g=open('Gita.txt')
9
+ #Gita=g.read()
10
+ loader = TextLoader('Gita.txt')
11
+ documents = loader.load()
12
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
13
+ texts = text_splitter.split_text(documents)
14
  docsearch = Chroma.from_texts(texts, embeddings)
15
  def answer(query):
16
  docs = docsearch.similarity_search(query)