Update app.py
Browse files
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 |
|