haxerwddle commited on
Commit
7a489b1
·
1 Parent(s): 4246f90

Model prompt change

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -36,7 +36,7 @@ pipe = pipeline(
36
  model=chat_model,
37
  tokenizer=tokenizer,
38
  device_map="auto",
39
- max_new_tokens=100
40
  )
41
 
42
  def clean_chat_output(full_text):
@@ -56,7 +56,7 @@ def explain_recycling(class_label):
56
  "content": (
57
  "You are an expert in waste sorting. "
58
  "You ALWAYS answer using exactly two bullet points:\n"
59
- "• Recycling type: <one short category>\n"
60
  "• Disposal: <clear, detailed correct sentence>\n"
61
  "No extra text, no introductions, no explanations."
62
  )
@@ -75,7 +75,7 @@ def explain_recycling(class_label):
75
 
76
  outputs = pipe(
77
  prompt,
78
- max_new_tokens=60,
79
  do_sample=False
80
  )
81
 
 
36
  model=chat_model,
37
  tokenizer=tokenizer,
38
  device_map="auto",
39
+ max_new_tokens=80
40
  )
41
 
42
  def clean_chat_output(full_text):
 
56
  "content": (
57
  "You are an expert in waste sorting. "
58
  "You ALWAYS answer using exactly two bullet points:\n"
59
+ "• Recycling type: <Item category>\n"
60
  "• Disposal: <clear, detailed correct sentence>\n"
61
  "No extra text, no introductions, no explanations."
62
  )
 
75
 
76
  outputs = pipe(
77
  prompt,
78
+ max_new_tokens=80,
79
  do_sample=False
80
  )
81