himanshukumar378 commited on
Commit
dc636e4
·
verified ·
1 Parent(s): c6246b4

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +8 -1
src/streamlit_app.py CHANGED
@@ -99,7 +99,14 @@ def get_vectorstore(text_chunks):
99
 
100
  def get_conversation_chain(vetorestore):
101
  handler = StdOutCallbackHandler()
102
- llm = HuggingFaceHub(repo_id="google/flan-t5-large", model_kwargs={"temperature":5,"max_length":64})
 
 
 
 
 
 
 
103
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
104
  conversation_chain = ConversationalRetrievalChain.from_llm(
105
  llm=llm,
 
99
 
100
  def get_conversation_chain(vetorestore):
101
  handler = StdOutCallbackHandler()
102
+ llm = HuggingFaceHub(
103
+ repo_id="google/flan-t5-large",
104
+ task="text2text-generation",
105
+ model_kwargs={
106
+ "temperature": 0.5,
107
+ "max_length": 512
108
+ }
109
+ )
110
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
111
  conversation_chain = ConversationalRetrievalChain.from_llm(
112
  llm=llm,