readme updated
Browse files
README.md
CHANGED
@@ -7,9 +7,11 @@ tags:
|
|
7 |
|
8 |
# text2tags-it
|
9 |
|
10 |
-
The model has been trained on a collection of 11k articles with tags. Its purpose is to create tags suitable for the given article.
|
11 |
|
12 |
-
|
|
|
|
|
13 |
|
14 |
```python
|
15 |
from transformers import T5ForConditionalGeneration,T5Tokenizer
|
@@ -39,4 +41,13 @@ def tag(text: str):
|
|
39 |
|
40 |
tags = tag(article)
|
41 |
print(tags)
|
42 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
# text2tags-it
|
9 |
|
10 |
+
The model has been trained on a collection of 11k news articles with tags. Its purpose is to create tags suitable for the given article.
|
11 |
|
12 |
+
### Usage
|
13 |
+
|
14 |
+
Sample code with an article from IlPost:
|
15 |
|
16 |
```python
|
17 |
from transformers import T5ForConditionalGeneration,T5Tokenizer
|
|
|
41 |
|
42 |
tags = tag(article)
|
43 |
print(tags)
|
44 |
+
```
|
45 |
+
|
46 |
+
### Overview
|
47 |
+
|
48 |
+
- Model: T5 ([it5-small](https://huggingface.co/gsarti/it5-small))
|
49 |
+
- Language: Italian
|
50 |
+
- Downstream-task: Summarization (for topic tagging)
|
51 |
+
- Training data: Custom dataset
|
52 |
+
- Code: See example
|
53 |
+
- Infrastructure: 1x T4
|