haxerwddle commited on
Commit
bc16505
·
1 Parent(s): 447e35f

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +1 -27
app.py CHANGED
@@ -40,33 +40,7 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
- Example 1:
44
- • Recycling type: Glass\n
45
- • Disposal: Rinse and put in the municipal glass-only bin\n
46
-
47
- Example 2:
48
- • Recycling type: Paper\n
49
- • Disposal: Place in the clean paper recycling stream\n
50
-
51
- Example 3:
52
- • Recycling type: Cardboard\n
53
- • Disposal: Flatten and place in the cardboard/boxboard bin\n
54
-
55
- Example 4:
56
- • Recycling type: Plastic (#1–#2 typically)\n
57
- • Disposal: Empty and place in your plastics recycling bin\n
58
-
59
- Example 5:
60
- • Recycling type: Metal (aluminum)\n
61
- • Disposal: Rinse and place in the municipal metal/aluminum bin\n
62
-
63
- Example 6:
64
- • Recycling type: Trash (non-recyclable)\n
65
- • Disposal: Put in general landfill waste bin\n
66
-
67
- Now answer using the examples:
68
- ALWAYS include "\n" at the end of every bullet.
69
- Item: {class_label}
70
  """
71
 
72
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
+ Give recycling tips for the item {class_label}.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
  """
45
 
46
  inputs = tokenizer(prompt, return_tensors="pt").input_ids