Marcin-XStudio
/

nu-extract-2-fork

Text Generation

Model card Files Files and versions

HowWebWorks commited on May 29, 2025

Commit

ca9c9ce

·

1 Parent(s): e2064f1

rename folder

Files changed (1) hide show

load_model.py +2 -2

load_model.py CHANGED Viewed

@@ -1,6 +1,6 @@
 from transformers import AutoModel, AutoTokenizer
 def load_custom_model():
-    tokenizer = AutoTokenizer.from_pretrained("./nu-extract-2-fork")
-    model = AutoModel.from_pretrained("./nu-extract-2-fork")
     return tokenizer, model

 from transformers import AutoModel, AutoTokenizer
 def load_custom_model():
+    tokenizer = AutoTokenizer.from_pretrained("./model")
+    model = AutoModel.from_pretrained("./model")
     return tokenizer, model