haxerwddle commited on
Commit
4c4bef2
·
1 Parent(s): f8bcafb

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ def explain_recycling(class_label):
49
  2. **How {class_label} is recycled or processed**
50
  3. **Optional tips for reducing waste or reusing {class_label}**
51
 
52
- Write the response in short paragraphs.
53
  """
54
 
55
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
@@ -60,7 +60,7 @@ def explain_recycling(class_label):
60
  do_sample=True,
61
  top_p = 0.9,)
62
 
63
- return tokenizer.decode(outputs[0], skip_special_tokens=True)
64
 
65
  # ------------------ PIPELINE ------------------
66
  def full_pipeline(image):
 
49
  2. **How {class_label} is recycled or processed**
50
  3. **Optional tips for reducing waste or reusing {class_label}**
51
 
52
+ Write the response in 3 paragraphs.
53
  """
54
 
55
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
60
  do_sample=True,
61
  top_p = 0.9,)
62
 
63
+ return tokenizer.decode(outputs[0])
64
 
65
  # ------------------ PIPELINE ------------------
66
  def full_pipeline(image):