PebinAPJ commited on
Commit
07d180f
·
verified ·
1 Parent(s): 24ceb44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def get_conversation_chain(vectorstore):
50
  llm = HuggingFaceEndpoint(
51
  repo_id="google/flan-t5-base",
52
  temperature=0.5,
53
- model_kwargs={"max_new_tokens": 248}
54
  )
55
 
56
  memory = ConversationBufferMemory(
 
50
  llm = HuggingFaceEndpoint(
51
  repo_id="google/flan-t5-base",
52
  temperature=0.5,
53
+ max_new_tokens=248
54
  )
55
 
56
  memory = ConversationBufferMemory(