AddieFoote commited on
Commit
bee25f4
·
1 Parent(s): 1c176df
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ else:
14
 
15
  def generate_response(prompt):
16
  inputs = tokenizer(prompt, return_tensors="pt")
17
- outputs = model.generate(**inputs, max_new_tokens=16, temperature=1.0)
18
  input_length = inputs['input_ids'].shape[1]
19
  new_token_ids = outputs[0][input_length:]
20
  new_tokens = tokenizer.decode(new_token_ids, skip_special_tokens=False)
 
14
 
15
  def generate_response(prompt):
16
  inputs = tokenizer(prompt, return_tensors="pt")
17
+ outputs = model.generate(**inputs, max_new_tokens=16, temperature=.7)
18
  input_length = inputs['input_ids'].shape[1]
19
  new_token_ids = outputs[0][input_length:]
20
  new_tokens = tokenizer.decode(new_token_ids, skip_special_tokens=False)