HowWebWorks commited on
Commit
0abee29
·
1 Parent(s): 93fe49d

put model into folder, add load_model.py

Browse files
Files changed (2) hide show
  1. load_model.py +6 -0
  2. nu-extract-2-fork +1 -0
load_model.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ from transformers import AutoModel, AutoTokenizer
2
+
3
+ def load_custom_model():
4
+ tokenizer = AutoTokenizer.from_pretrained("./nu-extract-2-fork")
5
+ model = AutoModel.from_pretrained("./nu-extract-2-fork")
6
+ return tokenizer, model
nu-extract-2-fork ADDED
@@ -0,0 +1 @@
 
 
1
+ Subproject commit 9a6475ba39b0798d3b8c8e0a3c4e099e29954761