Update README.md
Browse files
README.md
CHANGED
@@ -7,6 +7,9 @@ metrics:
|
|
7 |
library_name: transformers
|
8 |
pipeline_tag: text2text-generation
|
9 |
---
|
|
|
|
|
|
|
10 |
category_mapping = {
|
11 |
'Political':1,
|
12 |
'Economy':2,
|
@@ -18,3 +21,4 @@ category_mapping = {
|
|
18 |
'Accidents':8
|
19 |
# Add more categories as needed
|
20 |
}
|
|
|
|
7 |
library_name: transformers
|
8 |
pipeline_tag: text2text-generation
|
9 |
---
|
10 |
+
This model is under trial.
|
11 |
+
|
12 |
+
The number in the generated text represents the category of the news, as shown below.
|
13 |
category_mapping = {
|
14 |
'Political':1,
|
15 |
'Economy':2,
|
|
|
21 |
'Accidents':8
|
22 |
# Add more categories as needed
|
23 |
}
|
24 |
+
|