Spaces:
Paused
Paused
Updated app
Browse files
app.py
CHANGED
|
@@ -79,7 +79,7 @@ def generate(req: PromptRequest):
|
|
| 79 |
|
| 80 |
try:
|
| 81 |
# Limit tokens to avoid huge outputs
|
| 82 |
-
max_tokens =
|
| 83 |
|
| 84 |
output = pipe(req.prompt, max_new_tokens=max_tokens, do_sample=True)
|
| 85 |
text = output[0].get("generated_text", "").strip()
|
|
|
|
| 79 |
|
| 80 |
try:
|
| 81 |
# Limit tokens to avoid huge outputs
|
| 82 |
+
max_tokens = 150
|
| 83 |
|
| 84 |
output = pipe(req.prompt, max_new_tokens=max_tokens, do_sample=True)
|
| 85 |
text = output[0].get("generated_text", "").strip()
|