Spaces:
Runtime error
Runtime error
Commit
·
65dc784
1
Parent(s):
24f0d06
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,23 @@
|
|
1 |
import gradio as gr
|
|
|
|
|
|
|
|
|
2 |
from langchain.document_loaders import TextLoader
|
3 |
-
|
4 |
-
|
5 |
-
loader = TextLoader(
|
6 |
documents = loader.load()
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
def answer(query):
|
8 |
-
|
9 |
-
|
|
|
|
|
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()
|