haxerwddle commited on
Commit
9bd9b97
·
1 Parent(s): 87e3d93

prompt model change

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ chat_model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-base")
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
  Item: {class_label}
44
- Provide a short recycling instructions. No repeating.
45
  """
46
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
47
 
 
41
  def explain_recycling(class_label):
42
  prompt = f"""
43
  Item: {class_label}
44
+ Provide a recycling instructions for Item. No repeating.
45
  """
46
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
47