suolyer commited on
Commit
82648f7
•
1 Parent(s): 307ac88

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -52,10 +52,10 @@ Run the code
52
 
53
  ```python
54
  import argparse
55
- from fengshen import UbertPiplines
56
 
57
  total_parser = argparse.ArgumentParser("TASK NAME")
58
- total_parser = UbertPiplines.piplines_args(total_parser)
59
  args = total_parser.parse_args()
60
 
61
  args.pretrained_model_path = "IDEA-CCNL/Erlangshen-Ubert-330M-Chinese"
@@ -72,7 +72,7 @@ test_data=[
72
  "id": 0}
73
  ]
74
 
75
- model = UbertPiplines(args)
76
  result = model.predict(test_data)
77
  for line in result:
78
  print(line)
 
52
 
53
  ```python
54
  import argparse
55
+ from fengshen import UbertPipelines
56
 
57
  total_parser = argparse.ArgumentParser("TASK NAME")
58
+ total_parser = UbertPipelines.pipelines_args(total_parser)
59
  args = total_parser.parse_args()
60
 
61
  args.pretrained_model_path = "IDEA-CCNL/Erlangshen-Ubert-330M-Chinese"
 
72
  "id": 0}
73
  ]
74
 
75
+ model = UbertPipelines(args)
76
  result = model.predict(test_data)
77
  for line in result:
78
  print(line)