Woziii commited on
Commit
45f8781
·
verified ·
1 Parent(s): cd6f17c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def generate_text(input_text, temperature, top_p, top_k):
144
  with torch.no_grad():
145
  outputs = model.generate(
146
  **inputs,
147
- max_new_tokens=1,
148
  temperature=temperature,
149
  top_p=top_p,
150
  top_k=top_k
 
144
  with torch.no_grad():
145
  outputs = model.generate(
146
  **inputs,
147
+ max_new_tokens=10,
148
  temperature=temperature,
149
  top_p=top_p,
150
  top_k=top_k