ali121300 commited on
Commit
e245ef9
·
verified ·
1 Parent(s): 9cd6d7f

Update app2.py

Browse files
Files changed (1) hide show
  1. app2.py +1 -1
app2.py CHANGED
@@ -51,7 +51,7 @@ db=build_model(documents,chunk_size,chunk_overlap)
51
 
52
  ##################################################################################################
53
  def load_model(query,db):
54
- llm=HuggingFaceHub(repo_id="https://huggingface.co/spaces/ali121300/st_2/blob/main/all-MiniLM-L6-v2",repo_name="all-MiniLM-L6-v2-f16.gguf", model_kwargs={"temperature":0.1, "max_length":400})
55
  chain = load_qa_chain(llm, chain_type="stuff")
56
  docs = db.similarity_search(query)
57
  out=chain.run(input_documents=docs, question=query)
 
51
 
52
  ##################################################################################################
53
  def load_model(query,db):
54
+ llm=HuggingFaceHub(repo_id="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={"temperature":0.1, "max_length":400})
55
  chain = load_qa_chain(llm, chain_type="stuff")
56
  docs = db.similarity_search(query)
57
  out=chain.run(input_documents=docs, question=query)