Update app2.py
Browse files
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.
|
| 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 |
##############################################################################################
|