Update README.md
Browse files
README.md
CHANGED
|
@@ -42,6 +42,7 @@ To use the model:
|
|
| 42 |
|
| 43 |
```python
|
| 44 |
import json
|
|
|
|
| 45 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 46 |
|
| 47 |
def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):
|
|
|
|
| 42 |
|
| 43 |
```python
|
| 44 |
import json
|
| 45 |
+
import torch
|
| 46 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 47 |
|
| 48 |
def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):
|