mdacampora commited on
Commit
498a9d4
·
1 Parent(s): 830e8ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ model = PeftModel.from_pretrained(model, peft_model_id)
17
 
18
  def make_inference(problem):
19
  batch = tokenizer(
20
- f"User: {problem}",
21
  return_tensors="pt",
22
  )
23
 
 
17
 
18
  def make_inference(problem):
19
  batch = tokenizer(
20
+ problem,
21
  return_tensors="pt",
22
  )
23