batterydata commited on
Commit
a5fb832
1 Parent(s): 3082a1b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -8,7 +8,7 @@ datasets:
8
  metrics: squad
9
  ---
10
 
11
- # BERT-base-cased for QA
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-uncased-squad-v1"
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 = {