sayef
/

fsner-bert-base-uncased

Feature Extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

saif commited on Mar 28, 2022

Commit

cd3989a

·

1 Parent(s): ba83523

Update readme

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -74,11 +74,11 @@ support_texts = {
 device = 'cpu'
-tokenizer = FSNERTokenizerUtils("checkpoints/model")
 queries = tokenizer.tokenize(query_texts).to(device)
 supports = tokenizer.tokenize(list(support_texts.values())).to(device)
-model = FSNERModel("checkpoints/model")
 model.to(device)
 p_starts, p_ends = model.predict(queries, supports)

 device = 'cpu'
+tokenizer = FSNERTokenizerUtils("sayef/fsner-bert-base-uncased")
 queries = tokenizer.tokenize(query_texts).to(device)
 supports = tokenizer.tokenize(list(support_texts.values())).to(device)
+model = FSNERModel("sayef/fsner-bert-base-uncased")
 model.to(device)
 p_starts, p_ends = model.predict(queries, supports)