AdamyaG commited on
Commit
c8e3ace
1 Parent(s): a936bca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ def get_vectorstore_from_url(url):
32
  return vector_store
33
 
34
  def get_context_retriever_chain(vector_store):
35
- llm = HuggingFaceHub(repo_id = "google/gemma-2-9b", model_kwargs = {"temperature":0.5, "max_length":512})
36
  retriever = vector_store.as_retriever()
37
 
38
  prompt = ChatPromptTemplate.from_messages([
@@ -46,7 +46,7 @@ def get_context_retriever_chain(vector_store):
46
  return retriever_chain
47
 
48
  def get_conversational_rag_chain(retriever_chain):
49
- llm = HuggingFaceHub(repo_id = "google/gemma-2-9b", model_kwargs = {"temperature":0.5, "max_length":512})
50
 
51
  prompt = ChatPromptTemplate.from_messages([
52
  ("system", "Answer the user's questions based on the below context:\n\n{context}"),
 
32
  return vector_store
33
 
34
  def get_context_retriever_chain(vector_store):
35
+ llm = HuggingFaceHub(repo_id = "HuggingFaceH4/zephyr-7b-beta", model_kwargs = {"temperature":0.5, "max_length":512})
36
  retriever = vector_store.as_retriever()
37
 
38
  prompt = ChatPromptTemplate.from_messages([
 
46
  return retriever_chain
47
 
48
  def get_conversational_rag_chain(retriever_chain):
49
+ llm = HuggingFaceHub(repo_id = "HuggingFaceH4/zephyr-7b-beta", model_kwargs = {"temperature":0.5, "max_length":512})
50
 
51
  prompt = ChatPromptTemplate.from_messages([
52
  ("system", "Answer the user's questions based on the below context:\n\n{context}"),