Hugging Face's logo

HeyLucasLeao
/
byt5-small-pt-product-reviews

Text2Text Generation
Transformers
PyTorch
t5
text-generation-inference
Inference Endpoints
Model card Files Community
1
HeyLucasLeao commited on
Commit
a14de36
•
1 Parent(s): ec5f659

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -41,7 +41,7 @@ My true intention was totally educational, thus making available a this version
41
 
42
  How to use
43
  ``` python
44
- from transformers import AutoTokenizer, AutoModelForCausalLM
45
  import torch
46
 
47
  if torch.cuda.is_available():
@@ -51,8 +51,7 @@ else:
51
  print(device)
52
 
53
  tokenizer = AutoTokenizer.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
54
-
55
- model = AutoModelForCausalLM.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
56
  model.to(device)
57
 
58
  def classificar_review(review):
 
41
 
42
  How to use
43
  ``` python
44
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
45
  import torch
46
 
47
  if torch.cuda.is_available():
 
51
  print(device)
52
 
53
  tokenizer = AutoTokenizer.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
54
+ model = AutoModelForSeq2SeqLM.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
 
55
  model.to(device)
56
 
57
  def classificar_review(review):