Update README.md
Browse files
README.md
CHANGED
@@ -52,10 +52,10 @@ Run the code
|
|
52 |
|
53 |
```python
|
54 |
import argparse
|
55 |
-
from fengshen import
|
56 |
|
57 |
total_parser = argparse.ArgumentParser("TASK NAME")
|
58 |
-
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 =
|
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)
|