haxerwddle commited on
Commit
2bce3f2
·
1 Parent(s): 7f2b523

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
40
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
- Give short, 1 sentence recycling instructions for the item {class_label}.
44
  """
45
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
46
 
 
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