Aditya757864 commited on
Commit
5879dd8
·
verified ·
1 Parent(s): 3eeb1d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -26,6 +26,7 @@ def llm_pipeline():
26
  tokenizer = tokenizer,
27
  do_sample = True,
28
  temperature = 0.5,
 
29
  )
30
  local_llm = HuggingFacePipeline(pipeline=pipe)
31
  return local_llm
 
26
  tokenizer = tokenizer,
27
  do_sample = True,
28
  temperature = 0.5,
29
+ max_length = 300
30
  )
31
  local_llm = HuggingFacePipeline(pipeline=pipe)
32
  return local_llm