Chris4K commited on
Commit
af523e3
·
verified ·
1 Parent(s): eafad45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -705,7 +705,7 @@ Provide your suggestions in a Python dictionary format."""
705
  llm = HuggingFacePipeline.from_model_id(
706
  model_id="google/flan-t5-large",
707
  task="text2text-generation",
708
- model_kwargs={"temperature": 0.7, "max_new_tokens": 512}, # Changed max_length to max_new_tokens
709
  )
710
 
711
  # Generate suggestions
 
705
  llm = HuggingFacePipeline.from_model_id(
706
  model_id="google/flan-t5-large",
707
  task="text2text-generation",
708
+ model_kwargs={"temperature": 0.7, "max_length": 512}, # Changed max_length to max_new_tokens
709
  )
710
 
711
  # Generate suggestions