kaborg15 commited on
Commit
f2baf1b
1 Parent(s): ee4f217

Added temperature parameter to query

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,7 +14,9 @@ def query(payload):
14
  def get_completion(prompt):
15
  output = query({
16
  "inputs": f"{prompt}",
17
- "parameters": {}
 
 
18
  })
19
  return output[0]["generated_text"]
20
 
 
14
  def get_completion(prompt):
15
  output = query({
16
  "inputs": f"{prompt}",
17
+ "parameters": {
18
+ "temperature":0.6,
19
+ }
20
  })
21
  return output[0]["generated_text"]
22