doubledsbv commited on
Commit
8c4f787
1 Parent(s): 2734e1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -33,6 +33,10 @@ def chat_function(message, history, system_prompt,max_new_tokens,temperature):
33
  pipeline.tokenizer.eos_token_id,
34
  pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
35
  ]
 
 
 
 
36
  outputs = pipeline(
37
  prompt,
38
  max_new_tokens=max_new_tokens,
 
33
  pipeline.tokenizer.eos_token_id,
34
  pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
35
  ]
36
+
37
+ if temperature == 0:
38
+ temperature += 0.1
39
+
40
  outputs = pipeline(
41
  prompt,
42
  max_new_tokens=max_new_tokens,