ubermenchh commited on
Commit
3284c4f
1 Parent(s): 6dcf428

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -13
app.py CHANGED
@@ -25,19 +25,9 @@ def paper_changes(paper_id):
25
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=retriever, return_source_documents=True)
26
  return 'Ready!!'
27
 
28
- def paper_changes(paper_id):
29
- paper = next(arxiv.Client().results(arxiv.Search(id_list=[paper_id])))
30
- loader = PyPDFLoader(paper.download_pdf())
31
- documents = loader.load_and_split()
32
- text_splitter = CharacterTextSplitter(chunk_size=1024, chunk_overlap=0)
33
- texts = text_splitter.split_documents(documents)
34
- embeddings = HuggingFaceHubEmbeddings()
35
- db = Chroma.from_documents(texts, embeddings)
36
- retriever = db.as_retriever()
37
- llm = HuggingFaceHub(repo_id='mistralai/Mistral-7B-v0.1', model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096})
38
- global qa
39
- qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=retriever, return_source_documents=True)
40
- return 'Ready!!'
41
 
42
  def bot(history):
43
  response = infer(history[-1][0])
 
25
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type='stuff', retriever=retriever, return_source_documents=True)
26
  return 'Ready!!'
27
 
28
+ def add_text(history, text):
29
+ history = history + [(text, None)]
30
+ return history, ''
 
 
 
 
 
 
 
 
 
 
31
 
32
  def bot(history):
33
  response = infer(history[-1][0])