haxerwddle commited on
Commit
3dde239
·
1 Parent(s): 8fe984f

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -41,9 +41,9 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
  Give recycling instructions for item {class_label}.
44
- You should answer in this format:
45
- Recycling type: <trash type> \n
46
- Disposal: <At least 2 sentences, creative to dispose of the item>\n
47
  """
48
 
49
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
  Give recycling instructions for item {class_label}.
44
+ You should:
45
+ - Suggest ways to dispose of the item>\n
46
+ -Answer in 1 - 2 sentences.
47
  """
48
 
49
  inputs = tokenizer(prompt, return_tensors="pt").input_ids