sudhir2016 commited on
Commit
7e2ebdd
·
1 Parent(s): 82ce63f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -7,4 +7,16 @@ from langchain.document_loaders import TextLoader
7
  from langchain.chains import RetrievalQA
8
  import os
9
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = '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'
10
- llm = HuggingFaceHub(repo_id="google/flan-t5-xxl")
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  from langchain.chains import RetrievalQA
8
  import os
9
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = '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'
10
+ llm = HuggingFaceHub(repo_id="google/flan-t5-xxl")
11
+ B=open('us.txt')
12
+ Biden=B.read()
13
+ loader = TextLoader(Biden)
14
+ text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
15
+ texts = text_splitter.split_documents(documents)
16
+ db = Chroma.from_documents(texts, embeddings)
17
+ qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever)
18
+ def answer(query):
19
+ out=qa.run(query)
20
+ return out
21
+ demo = gr.Interface(fn=answer, inputs='text',outputs='text',examples=[['What did the president say about Ketanji Jackson Brown']])
22
+ demo.launch()