captain-awesome commited on
Commit
2da7358
1 Parent(s): 14df33b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -137,17 +137,17 @@ 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="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="lmsys/fastchat-t5-3b-v1.0",
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)
 
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="lmsys/fastchat-t5-3b-v1.0",
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)