Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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
|