captain-awesome commited on
Commit
b63633d
·
verified ·
1 Parent(s): 77432f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -137,6 +137,11 @@ def get_response(user_input):
137
  task="text2text-generation",
138
  # model_kwargs={"temperature": 0.2},
139
  )
 
 
 
 
 
140
  retriever_chain = get_context_retriever_chain(st.session_state.vector_store,llm)
141
  conversation_rag_chain = get_conversational_rag_chain(retriever_chain,llm)
142
 
 
137
  task="text2text-generation",
138
  # model_kwargs={"temperature": 0.2},
139
  )
140
+ llm = HuggingFacePipeline.from_model_id(
141
+ model_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
142
+ task="text2text-generation",
143
+ # model_kwargs={"temperature": 0.2},
144
+ )
145
  retriever_chain = get_context_retriever_chain(st.session_state.vector_store,llm)
146
  conversation_rag_chain = get_conversational_rag_chain(retriever_chain,llm)
147