nguyennghia0902 commited on
Commit
8e77477
1 Parent(s): 0b0bdbf

Update SampleQA.py

Browse files
Files changed (1) hide show
  1. SampleQA.py +1 -1
SampleQA.py CHANGED
@@ -5,7 +5,7 @@ import random
5
  from transformers import ElectraTokenizerFast, TFElectraForQuestionAnswering
6
  from datasets import Dataset, DatasetDict, load_dataset
7
 
8
- model_hf = "nguyennghia0902/electra-small-discriminator_0.0005_32"
9
  tokenizer = ElectraTokenizerFast.from_pretrained(model_hf)
10
  reload_model = TFElectraForQuestionAnswering.from_pretrained(model_hf)
11
 
 
5
  from transformers import ElectraTokenizerFast, TFElectraForQuestionAnswering
6
  from datasets import Dataset, DatasetDict, load_dataset
7
 
8
+ model_hf = "nguyennghia0902/electra-small-discriminator_0.0001_16_15e"
9
  tokenizer = ElectraTokenizerFast.from_pretrained(model_hf)
10
  reload_model = TFElectraForQuestionAnswering.from_pretrained(model_hf)
11