Update app.py

app.py

@@ -22,7 +22,7 @@ def make_inference(product, description):
     output_tokens = model.generate(**batch, max_new_tokens=200)
   return tokenizer.decode(output_tokens[0], skip_special_tokens=True)
 '''
-def make_inference(product_name, product_description):
+def make_inference(product, description):
     batch = tokenizer(f"### INSTRUCTION\nBelow is a product and description, please write a marketing email for this product.\n\n### Product:\n{product}\n### Description:\n{description}\n\n### Marketing Email:\n", return_tensors='pt')
 
     batch = {key: value.to('cuda:0') for key, value in batch.items()}