Commit
•
0a89656
1
Parent(s):
3d54c9a
Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
---
|
| 2 |
|
|
|
|
| 3 |
license: mit
|
| 4 |
datasets:
|
| 5 |
- hotchpotch/JQaRA
|
|
@@ -59,7 +60,7 @@ scores = model.predict([(query, passage) for passage in passages])
|
|
| 59 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 60 |
from torch.nn import Sigmoid
|
| 61 |
|
| 62 |
-
MODEL_NAME = "hotchpotch/japanese-reranker-cross-encoder-xsmall-v1
|
| 63 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 64 |
|
| 65 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
|
|
|
| 1 |
---
|
| 2 |
|
| 3 |
+
|
| 4 |
license: mit
|
| 5 |
datasets:
|
| 6 |
- hotchpotch/JQaRA
|
|
|
|
| 60 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 61 |
from torch.nn import Sigmoid
|
| 62 |
|
| 63 |
+
MODEL_NAME = "hotchpotch/japanese-reranker-cross-encoder-xsmall-v1"
|
| 64 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 65 |
|
| 66 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|