sudhir2016 commited on
Commit
65dc784
·
1 Parent(s): 24f0d06

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -5
app.py CHANGED
@@ -1,11 +1,23 @@
1
  import gradio as gr
 
 
 
 
2
  from langchain.document_loaders import TextLoader
3
- g=open('test.txt')
4
- g1=g.read()
5
- loader = TextLoader('test.txt')
6
  documents = loader.load()
 
 
 
 
 
 
7
  def answer(query):
8
- out=g1
9
- return out
 
 
10
  demo = gr.Interface(fn=answer, inputs='text',outputs='text',examples=[['song celestial']])
11
  demo.launch()
 
1
  import gradio as gr
2
+ from langchain.vectorstores import Qdrant
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
+ Gita=open('Gita.txt')
9
+ loader = TextLoader(Gita)
10
  documents = loader.load()
11
+ text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
12
+ docs = text_splitter.split_text(documents)
13
+ qdrant = Qdrant.from_documents(
14
+ docs, embeddings,
15
+ location=":memory:",
16
+ collection_name="my_documents",)
17
  def answer(query):
18
+ out = qdrant.similarity_search_with_score(query)
19
+ out1=out[0]
20
+ out2=out1[0].page_content
21
+ return out2
22
  demo = gr.Interface(fn=answer, inputs='text',outputs='text',examples=[['song celestial']])
23
  demo.launch()