HeyLucasLeao
/

byt5-base-pt-product-reviews

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

HeyLucasLeao commited on Jul 9, 2021

Commit

c6b367f

•

1 Parent(s): 50bf411

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -50,8 +50,8 @@ else:
     device = torch.device('cpu')
 print(device)
-tokenizer = AutoTokenizer.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
-model = AutoModelForSeq2SeqLM.from_pretrained("HeyLucasLeao/byt5-small-pt-product-reviews")
 model.to(device)
 def classificar_review(review):

     device = torch.device('cpu')
 print(device)
+tokenizer = AutoTokenizer.from_pretrained("HeyLucasLeao/byt5-base-pt-product-reviews")
+model = AutoModelForSeq2SeqLM.from_pretrained("HeyLucasLeao/byt5-base-pt-product-reviews")
 model.to(device)
 def classificar_review(review):