Abaryan commited on
Commit
4a74617
·
verified ·
1 Parent(s): 8983f65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=256,
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,