diginoron commited on
Commit
c7083c4
·
verified ·
1 Parent(s): 6cf8230

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ def rewrite_answer(question, retrieved_answer):
57
  inputs = tokenizer(prompt, return_tensors="pt").to(device)
58
  outputs = model.generate(
59
  **inputs,
60
- max_new_tokens=64,
61
  temperature=0.7,
62
  do_sample=True,
63
  top_p=0.9
 
57
  inputs = tokenizer(prompt, return_tensors="pt").to(device)
58
  outputs = model.generate(
59
  **inputs,
60
+ max_new_tokens=96,
61
  temperature=0.7,
62
  do_sample=True,
63
  top_p=0.9