Chris Alexiuk commited on
Commit
faa0b42
1 Parent(s): 6697fea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
15
  # Load the Lora model
16
  model = PeftModel.from_pretrained(model, peft_model_id)
17
 
18
- def make_inference(product, description):
19
  batch = tokenizer(
20
  f"Please answer the following question to the best of your ability.\n\n### Question:\n{question}\n### Answer:\n",
21
  return_tensors="pt",
 
15
  # Load the Lora model
16
  model = PeftModel.from_pretrained(model, peft_model_id)
17
 
18
+ def make_inference(question):
19
  batch = tokenizer(
20
  f"Please answer the following question to the best of your ability.\n\n### Question:\n{question}\n### Answer:\n",
21
  return_tensors="pt",