Chan Lee commited on
Commit
a8cb60d
1 Parent(s): 476cd9d
app.py CHANGED
@@ -4,7 +4,7 @@ from gradio.mix import Parallel
4
 
5
  predictor_bert = ktrain.load_predictor('bert')
6
  predictor_mental = ktrain.load_predictor('mentalbert')
7
- predictor_phs = ktrain.load_predictor('phs-bert')
8
 
9
  def func_bert(text):
10
  results = predictor_bert.predict(str(text))
@@ -19,11 +19,11 @@ def func_phs(text):
19
  return str(results)
20
 
21
  # bert_io = gr.Interface(fn=func_bert, inputs="text", outputs="text")
22
- mental_io = gr.Interface(fn=func_mental, inputs="text", outputs="text")
23
- # phs_io = gr.Interface(fn=func_phs, inputs="text", outputs="text")
24
 
25
  # Parallel(bert_io, mental_io, phs_io).launch()
26
 
27
  # bert_io.launch()
28
- mental_io.launch()
29
- # phs_io.launch()
 
4
 
5
  predictor_bert = ktrain.load_predictor('bert')
6
  predictor_mental = ktrain.load_predictor('mentalbert')
7
+ predictor_phs = ktrain.load_predictor('phsbert')
8
 
9
  def func_bert(text):
10
  results = predictor_bert.predict(str(text))
 
19
  return str(results)
20
 
21
  # bert_io = gr.Interface(fn=func_bert, inputs="text", outputs="text")
22
+ # mental_io = gr.Interface(fn=func_mental, inputs="text", outputs="text")
23
+ phs_io = gr.Interface(fn=func_phs, inputs="text", outputs="text")
24
 
25
  # Parallel(bert_io, mental_io, phs_io).launch()
26
 
27
  # bert_io.launch()
28
+ # mental_io.launch()
29
+ phs_io.launch()
phsbert/config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "../input/ctvacbertlm",
3
+ "architectures": [
4
+ "BertForMaskedLM"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "gradient_checkpointing": false,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.1,
10
+ "hidden_size": 1024,
11
+ "id2label": {
12
+ "0": "LABEL_0",
13
+ "1": "LABEL_1",
14
+ "2": "LABEL_2"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 4096,
18
+ "label2id": {
19
+ "LABEL_0": 0,
20
+ "LABEL_1": 1,
21
+ "LABEL_2": 2
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "max_position_embeddings": 512,
25
+ "model_type": "bert",
26
+ "num_attention_heads": 16,
27
+ "num_hidden_layers": 24,
28
+ "pad_token_id": 0,
29
+ "position_embedding_type": "absolute",
30
+ "total_flos": 2232610412725615392,
31
+ "transformers_version": "4.3.3",
32
+ "type_vocab_size": 2,
33
+ "use_cache": true,
34
+ "vocab_size": 30522
35
+ }
phsbert/special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
phsbert/tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:258f3f004930a82f225766ede2e578c6b4e6cd8d743ccf018f3d92f73a1b6a2f
3
+ size 1341096820
phsbert/tf_model.preproc ADDED
Binary file (2.26 kB). View file
 
phsbert/tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"do_lower_case": true, "unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]", "tokenize_chinese_chars": true, "strip_accents": null, "special_tokens_map_file": null, "full_tokenizer_file": null, "name_or_path": "../input/ctvacbertlm", "do_basic_tokenize": true, "never_split": null}
phsbert/vocab.txt ADDED
The diff for this file is too large to render. See raw diff