mtyrrell commited on
Commit
5dc4473
1 Parent(s): 1e6395c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ retriever = EmbeddingRetriever(
97
  )
98
 
99
  # Initialize the PromptNode
100
- pn = PromptNode(model_name_or_path=model_name, default_prompt_template=template, api_key=openai_key, max_length=700)
101
 
102
  # Initialize the pipeline
103
  pipe = Pipeline()
 
97
  )
98
 
99
  # Initialize the PromptNode
100
+ pn = PromptNode(model_name_or_path=model_name, default_prompt_template=template, api_key=openai_key, max_length=700, model_kwargs={"generation_kwargs": {"do_sample": True, "temperature": 0}})
101
 
102
  # Initialize the pipeline
103
  pipe = Pipeline()