SankarV commited on
Commit
5388ef0
1 Parent(s): b063b93

Upload 3 files

Browse files

Initial question own documents.. first hugging face spaces application

Files changed (3) hide show
  1. app.py +35 -0
  2. guide1.txt +0 -0
  3. requirements.txt +6 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain.embeddings.openai import OpenAIEmbeddings
2
+ from langchain.vectorstores import Chroma
3
+ from langchain.text_splitter import CharacterTextSplitter
4
+ from langchain.chains.question_answering import load_qa_chain
5
+ from langchain.llms import OpenAI
6
+ import os
7
+ from dotenv import load_dotenv
8
+ import gradio as gr
9
+
10
+ load_dotenv()
11
+
12
+ with open("guide1.txt") as f:
13
+ hitchhikersguide = f.read()
14
+
15
+ text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0, separator = "\n")
16
+ texts = text_splitter.split_text(hitchhikersguide)
17
+
18
+ embeddings = OpenAIEmbeddings()
19
+ docsearch = Chroma.from_texts(texts, embeddings, metadatas=[{"source": str(i)} for i in range(len(texts))]).as_retriever()
20
+ chain = load_qa_chain(OpenAI(temperature=0), chain_type="stuff")
21
+
22
+ def make_inference(query):
23
+ docs = docsearch.get_relevant_documents(query)
24
+ return (chain.run(input_documents=docs, question=query))
25
+
26
+
27
+
28
+ if __name__ == "__main__":
29
+ # make a gradio interface
30
+ demo = gr.Interface(
31
+ fn = make_inference,inputs="text",outputs="text",
32
+ title="🗣️TalkToMyDoc📄",
33
+ description="🗣️TalkToMyDoc📄 is a tool that allows you to ask questions about a document. In this case - Hitch Hitchhiker's Guide to the Galaxy.",
34
+ )
35
+ demo.launch()
guide1.txt ADDED
The diff for this file is too large to render. See raw diff
 
requirements.txt ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ langchain
2
+ openai
3
+ tiktoken
4
+ chromadb
5
+ gradio
6
+ python-dotenv