Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
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
|