ali121300 commited on
Commit
8822641
·
verified ·
1 Parent(s): 9cc0225

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +1 -1
app2.py CHANGED
@@ -53,7 +53,7 @@ db=build_model(documents,chunk_size,chunk_overlap)
53
  def load_model(query,db):
54
  llm=HuggingFaceHub(repo_id="google/gemma-7b", model_kwargs={"temperature":0.1, "max_length":300})
55
  chain = load_qa_chain(llm, chain_type="stuff")
56
- docs = db.similarity_search_with_score(query)
57
  out=chain.run(input_documents=docs, question=query)
58
  return out
59
  ##############################################################################################
 
53
  def load_model(query,db):
54
  llm=HuggingFaceHub(repo_id="google/gemma-7b", model_kwargs={"temperature":0.1, "max_length":300})
55
  chain = load_qa_chain(llm, chain_type="stuff")
56
+ docs = db.similarity_search(query)
57
  out=chain.run(input_documents=docs, question=query)
58
  return out
59
  ##############################################################################################