batterydata
commited on
Commit
路
a5fb832
1
Parent(s):
3082a1b
Update README.md
Browse files
README.md
CHANGED
@@ -8,7 +8,7 @@ datasets:
|
|
8 |
metrics: squad
|
9 |
---
|
10 |
|
11 |
-
#
|
12 |
**Language model:** batteryonlybert-cased
|
13 |
**Language:** English
|
14 |
**Downstream-task:** Extractive QA
|
@@ -42,7 +42,7 @@ Evaluated on the battery device dataset.
|
|
42 |
```python
|
43 |
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
|
44 |
|
45 |
-
model_name = "batterydata/batteryonlybert-
|
46 |
# a) Get predictions
|
47 |
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
|
48 |
QA_input = {
|
|
|
8 |
metrics: squad
|
9 |
---
|
10 |
|
11 |
+
# BatteryOnlyBERT-cased for QA
|
12 |
**Language model:** batteryonlybert-cased
|
13 |
**Language:** English
|
14 |
**Downstream-task:** Extractive QA
|
|
|
42 |
```python
|
43 |
from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
|
44 |
|
45 |
+
model_name = "batterydata/batteryonlybert-cased-squad-v1"
|
46 |
# a) Get predictions
|
47 |
nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
|
48 |
QA_input = {
|