haxerwddle commited on
Commit
3f1a615
·
1 Parent(s): 2bce3f2

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -40,14 +40,13 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
- Recycling instruction for the item {class_label}.
44
- """
45
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
46
 
47
  outputs = chat_model.generate(
48
  inputs,
49
  max_length=80,
50
- do_sample=False,
51
  repetition_penalty=2.0,
52
  )
53
 
 
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
+ Give short recycling instructions for the item {class_label}."""
 
44
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
45
 
46
  outputs = chat_model.generate(
47
  inputs,
48
  max_length=80,
49
+ do_sample=False,
50
  repetition_penalty=2.0,
51
  )
52