loocorez commited on
Commit
6a81f43
·
verified ·
1 Parent(s): 79e4e0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def reverse_text(user_text: str) -> str:
32
  with torch.no_grad():
33
  out = model.generate(
34
  **inputs,
35
- max_new_tokens=256,
36
  do_sample=False, # <- deterministic
37
  temperature=0.0,
38
  pad_token_id=tokenizer.eos_token_id,
 
32
  with torch.no_grad():
33
  out = model.generate(
34
  **inputs,
35
+ max_new_tokens=2048,
36
  do_sample=False, # <- deterministic
37
  temperature=0.0,
38
  pad_token_id=tokenizer.eos_token_id,