Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -137,11 +137,18 @@ def get_response(user_input):
|
|
| 137 |
# task="text2text-generation",
|
| 138 |
# # model_kwargs={"temperature": 0.2},
|
| 139 |
# )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 140 |
llm = HuggingFacePipeline.from_model_id(
|
| 141 |
-
model_id="
|
| 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 |
|
|
|
|
| 137 |
# task="text2text-generation",
|
| 138 |
# # model_kwargs={"temperature": 0.2},
|
| 139 |
# )
|
| 140 |
+
# llm = HuggingFacePipeline.from_model_id(
|
| 141 |
+
# model_id="google-t5/t5-small",
|
| 142 |
+
# task="text2text-generation",
|
| 143 |
+
# # model_kwargs={"temperature": 0.2},
|
| 144 |
+
# )
|
| 145 |
+
|
| 146 |
llm = HuggingFacePipeline.from_model_id(
|
| 147 |
+
model_id="describeai/gemini-small",
|
| 148 |
task="text2text-generation",
|
| 149 |
# model_kwargs={"temperature": 0.2},
|
| 150 |
)
|
| 151 |
+
|
| 152 |
retriever_chain = get_context_retriever_chain(st.session_state.vector_store,llm)
|
| 153 |
conversation_rag_chain = get_conversational_rag_chain(retriever_chain,llm)
|
| 154 |
|