Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -169,8 +169,8 @@ if pdf:
|
|
| 169 |
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
|
| 170 |
output = model.generate(
|
| 171 |
**inputs,
|
| 172 |
-
max_new_tokens=
|
| 173 |
-
temperature=0.
|
| 174 |
repetition_penalty=1.2,
|
| 175 |
eos_token_id=tokenizer.eos_token_id,
|
| 176 |
)
|
|
|
|
| 169 |
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
|
| 170 |
output = model.generate(
|
| 171 |
**inputs,
|
| 172 |
+
max_new_tokens=500,
|
| 173 |
+
temperature=0.3,
|
| 174 |
repetition_penalty=1.2,
|
| 175 |
eos_token_id=tokenizer.eos_token_id,
|
| 176 |
)
|