haxerwddle commited on
Commit
2cdedfe
·
1 Parent(s): 7dbd5e9

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -43,7 +43,7 @@ def explain_recycling(class_label):
43
  You are a waste management expert.
44
 
45
  The waste item is classified as: **{class_label}**
46
- Provide a short explanation including:
47
 
48
  1. **How to dispose of {class_label} correctly**
49
  2. **How {class_label} is recycled or processed**
@@ -56,7 +56,8 @@ def explain_recycling(class_label):
56
  outputs = chat_model.generate(
57
  inputs,
58
  max_length=250,
59
- do_sample=False,)
 
60
 
61
  return tokenizer.decode(outputs[0])
62
 
 
43
  You are a waste management expert.
44
 
45
  The waste item is classified as: **{class_label}**
46
+ Provide a 3-paragraph explanation including:
47
 
48
  1. **How to dispose of {class_label} correctly**
49
  2. **How {class_label} is recycled or processed**
 
56
  outputs = chat_model.generate(
57
  inputs,
58
  max_length=250,
59
+ do_sample=True,
60
+ top_p = 0.9,)
61
 
62
  return tokenizer.decode(outputs[0])
63