haxerwddle commited on
Commit
0bd541d
·
1 Parent(s): 7a489b1

Model prompt change

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -76,7 +76,9 @@ def explain_recycling(class_label):
76
  outputs = pipe(
77
  prompt,
78
  max_new_tokens=80,
79
- do_sample=False
 
 
80
  )
81
 
82
  raw = outputs[0]["generated_text"]
 
76
  outputs = pipe(
77
  prompt,
78
  max_new_tokens=80,
79
+ do_sample=True,
80
+ top_p = 0.9,
81
+ temperature=0.3
82
  )
83
 
84
  raw = outputs[0]["generated_text"]