haxerwddle commited on
Commit
3fd6140
·
1 Parent(s): abc7f03

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -40,8 +40,10 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
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
 
 
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
+ Item: {class_label}
44
+ Provide a recycling description in exactly this format:
45
+ • Recycling type: <answer>\n
46
+ • Disposal: <answer>\n
47
  """
48
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
49