efederici commited on
Commit
8aca81d
1 Parent(s): 5cf5d93

readme updated

Browse files
Files changed (1) hide show
  1. README.md +14 -3
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
- Sample code with a IlPost article:
 
 
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