Update README.md
Browse files
README.md
CHANGED
|
@@ -60,6 +60,7 @@ To use the model:
|
|
| 60 |
|
| 61 |
```python
|
| 62 |
import json
|
|
|
|
| 63 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 64 |
|
| 65 |
def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):
|
|
|
|
| 60 |
|
| 61 |
```python
|
| 62 |
import json
|
| 63 |
+
import torch
|
| 64 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 65 |
|
| 66 |
def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):
|