numind
/

NuExtract-1.5

Text Generation

Inference Endpoints

Model card Files Files and versions Community

liamcripwell commited on about 12 hours ago

Commit

0d9cff7

•

1 Parent(s): 8c83d7b

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -60,6 +60,7 @@ To use the model:
 ```python
 import json
 from transformers import AutoModelForCausalLM, AutoTokenizer
 def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):

 ```python
 import json
+import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):