captain-awesome commited on
Commit
a7d84ba
·
verified ·
1 Parent(s): 98f8b95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -87,7 +87,7 @@ def get_response(user_input):
87
  llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
88
  llm = HuggingFaceHub(
89
  repo_id=llm_model,
90
- model_kwargs={"temperature": temperature, "max_new_tokens": 250, "top_k": top_k}
91
  )
92
  retriever_chain = get_context_retriever_chain(st.session_state.vector_store,llm)
93
  conversation_rag_chain = get_conversational_rag_chain(retriever_chain,llm)
 
87
  llm_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
88
  llm = HuggingFaceHub(
89
  repo_id=llm_model,
90
+ model_kwargs={"temperature": 0.3, "max_new_tokens": 250, "top_k": 3}
91
  )
92
  retriever_chain = get_context_retriever_chain(st.session_state.vector_store,llm)
93
  conversation_rag_chain = get_conversational_rag_chain(retriever_chain,llm)