captain-awesome commited on
Commit
ff23be6
1 Parent(s): b5a3ee0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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="google-t5/t5-small",
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