Abrar20 commited on
Commit
9626076
·
verified ·
1 Parent(s): bc7e478

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,7 +47,7 @@ def load_pdf_to_langchain(pdf_path, repo_id):
47
  retriever = db.as_retriever()
48
  llm = HuggingFaceHub(
49
  repo_id=repo_id,
50
- model_kwargs={'temperature': 0.5, 'max_new_tokens': 2096}
51
  )
52
  qa_chain = get_chain(llm, retriever)
53
  return qa_chain
@@ -95,7 +95,7 @@ with gr.Blocks(css=css, theme='Taithrah/Minimal') as demo:
95
  'RWKV/rwkv-4-7b-pile',
96
  # Add more models here as desired
97
  ],
98
- value='mistralai/Mistral-7B-Instruct-v0.1"
99
  )
100
 
101
  # Status display
 
47
  retriever = db.as_retriever()
48
  llm = HuggingFaceHub(
49
  repo_id=repo_id,
50
+ model_kwargs={'temperature': 0.3, 'max_new_tokens': 2096}
51
  )
52
  qa_chain = get_chain(llm, retriever)
53
  return qa_chain
 
95
  'RWKV/rwkv-4-7b-pile',
96
  # Add more models here as desired
97
  ],
98
+ value='mistralai/Mistral-7B-Instruct-v0.1'
99
  )
100
 
101
  # Status display