haxerwddle commited on
Commit
34f1bc1
·
1 Parent(s): 9ba5509

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -58,8 +58,7 @@ def explain_recycling(class_label):
58
  outputs = chat_model.generate(
59
  inputs,
60
  max_length=400,
61
- do_sample=True,
62
- top_p = 0.9,
63
  )
64
 
65
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
58
  outputs = chat_model.generate(
59
  inputs,
60
  max_length=400,
61
+ do_sample=False
 
62
  )
63
 
64
  return tokenizer.decode(outputs[0], skip_special_tokens=True)