haxerwddle commited on
Commit
abc7f03
·
1 Parent(s): 7d958d0

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -40,10 +40,9 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
 
43
  Give short recycling instructions for the item {class_label}.
44
- Rules:
45
- - NEVER repeat any sentence.
46
- - NEVER restate the same phrase more than once."""
47
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
48
 
49
  outputs = chat_model.generate(
 
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
+
44
  Give short recycling instructions for the item {class_label}.
45
+ """
 
 
46
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
47
 
48
  outputs = chat_model.generate(