legacy107 commited on
Commit
f48d6e2
1 Parent(s): c4569dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def paraphrase_answer(question, answer):
41
 
42
  # Generate the answer
43
  with torch.no_grad():
44
- generated_ids = paraphrase_model.generate(input_ids, max_new_tokens=max_target_length)
45
 
46
  # Decode and return the generated answer
47
  paraphrased_answer = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
41
 
42
  # Generate the answer
43
  with torch.no_grad():
44
+ generated_ids = paraphrase_model.generate(input_ids=input_ids, max_new_tokens=max_target_length)
45
 
46
  # Decode and return the generated answer
47
  paraphrased_answer = tokenizer.decode(generated_ids[0], skip_special_tokens=True)