axotion
/

polish-reranker-base-ranknet

sentence-transformers

text-classification

information-retrieval

Model card Files Files and versions Community

axotion commited on Mar 17

Commit

78efea0

·

verified ·

1 Parent(s): b2420a4

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ answers = [
 ]
 model = CrossEncoder(
-    "sdadas/polish-reranker-base-ranknet",
     default_activation_function=torch.nn.Identity(),
     max_length=512,
     device="cuda" if torch.cuda.is_available() else "cpu"
@@ -58,7 +58,7 @@ answers = [
     "Gdy trwała kampania politycy zapewniali, że rozprawią się z zakazem niedzielnego handlu."
 ]
-model_name = "sdadas/polish-reranker-base-ranknet"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 texts = [f"{query}</s></s>{answer}" for answer in answers]

 ]
 model = CrossEncoder(
+    "axotion/polish-reranker-base-ranknet",
     default_activation_function=torch.nn.Identity(),
     max_length=512,
     device="cuda" if torch.cuda.is_available() else "cpu"
     "Gdy trwała kampania politycy zapewniali, że rozprawią się z zakazem niedzielnego handlu."
 ]
+model_name = "axotion/polish-reranker-base-ranknet"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 texts = [f"{query}</s></s>{answer}" for answer in answers]