Spaces:
Running
Running
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,
|