Update app.py
Browse files
app.py
CHANGED
|
@@ -288,7 +288,6 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 288 |
|
| 289 |
|
| 290 |
tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract")
|
| 291 |
-
tokenizer.eos = tokenizer("<|end-output|>")
|
| 292 |
|
| 293 |
model.to("cuda")
|
| 294 |
model.eval()
|
|
|
|
| 288 |
|
| 289 |
|
| 290 |
tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract")
|
|
|
|
| 291 |
|
| 292 |
model.to("cuda")
|
| 293 |
model.eval()
|