Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,6 +36,7 @@ The following SQL query best answers the question `{user_question}`:
|
|
| 36 |
eos_token_id=tokenizer.eos_token_id,
|
| 37 |
pad_token_id=tokenizer.eos_token_id,
|
| 38 |
max_new_tokens=400,
|
|
|
|
| 39 |
num_beams=1,
|
| 40 |
)
|
| 41 |
outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
|
|
|
| 36 |
eos_token_id=tokenizer.eos_token_id,
|
| 37 |
pad_token_id=tokenizer.eos_token_id,
|
| 38 |
max_new_tokens=400,
|
| 39 |
+
do_sample=False,
|
| 40 |
num_beams=1,
|
| 41 |
)
|
| 42 |
outputs = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|