efederici commited on
Commit
58626c4
1 Parent(s): 5441fe4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -30,10 +30,10 @@ The model has been trained on a collection of 28k news articles with tags. Its p
30
  Sample code with an article from IlPost:
31
 
32
  ```python
33
- from transformers import T5ForConditionalGeneration,T5Tokenizer
34
 
35
- model = T5ForConditionalGeneration.from_pretrained("efederici/text2tags")
36
- tokenizer = T5Tokenizer.from_pretrained("efederici/text2tags")
37
 
38
  article = '''
39
  Da bambino era preoccupato che al mondo non ci fosse più nulla da scoprire. Ma i suoi stessi studi gli avrebbero dato torto: insieme a James Watson, nel 1953 Francis Crick strutturò il primo modello di DNA, la lunga sequenza di codici che identifica ogni essere vivente, rendendolo unico e diverso da tutti gli altri.
@@ -64,12 +64,12 @@ print(tags)
64
  Assuming paragraphs are divided by: '\n\n'.
65
 
66
  ```python
67
- from transformers import T5ForConditionalGeneration,T5Tokenizer
68
  import itertools
69
  import re
70
 
71
- model = T5ForConditionalGeneration.from_pretrained("efederici/text2tags")
72
- tokenizer = T5Tokenizer.from_pretrained("efederici/text2tags")
73
 
74
  article = '''
75
  Da bambino era preoccupato che al mondo non ci fosse più nulla da scoprire. Ma i suoi stessi studi gli avrebbero dato torto: insieme a James Watson, nel 1953 Francis Crick strutturò il primo modello di DNA, la lunga sequenza di codici che identifica ogni essere vivente, rendendolo unico e diverso da tutti gli altri.
 
30
  Sample code with an article from IlPost:
31
 
32
  ```python
33
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
34
 
35
+ model = AutoModelForSeq2SeqLM.from_pretrained("efederici/text2tags")
36
+ tokenizer = AutoTokenizer.from_pretrained("efederici/text2tags")
37
 
38
  article = '''
39
  Da bambino era preoccupato che al mondo non ci fosse più nulla da scoprire. Ma i suoi stessi studi gli avrebbero dato torto: insieme a James Watson, nel 1953 Francis Crick strutturò il primo modello di DNA, la lunga sequenza di codici che identifica ogni essere vivente, rendendolo unico e diverso da tutti gli altri.
 
64
  Assuming paragraphs are divided by: '\n\n'.
65
 
66
  ```python
67
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
68
  import itertools
69
  import re
70
 
71
+ model = AutoModelForSeq2SeqLM.from_pretrained("efederici/text2tags")
72
+ tokenizer = AutoTokenizer.from_pretrained("efederici/text2tags")
73
 
74
  article = '''
75
  Da bambino era preoccupato che al mondo non ci fosse più nulla da scoprire. Ma i suoi stessi studi gli avrebbero dato torto: insieme a James Watson, nel 1953 Francis Crick strutturò il primo modello di DNA, la lunga sequenza di codici che identifica ogni essere vivente, rendendolo unico e diverso da tutti gli altri.