Spaces:
Sleeping
Sleeping
Abaryan
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def predict(question: str, option_a: str, option_b: str, option_c: str, option_d
|
|
| 58 |
with torch.no_grad():
|
| 59 |
outputs = model.generate(
|
| 60 |
**inputs,
|
| 61 |
-
max_new_tokens=
|
| 62 |
temperature=0.6,
|
| 63 |
top_p=0.9,
|
| 64 |
do_sample=True,
|
|
|
|
| 58 |
with torch.no_grad():
|
| 59 |
outputs = model.generate(
|
| 60 |
**inputs,
|
| 61 |
+
max_new_tokens=20,
|
| 62 |
temperature=0.6,
|
| 63 |
top_p=0.9,
|
| 64 |
do_sample=True,
|