Spaces:

tclopess
/

MultiLabel-Classificando-Sinopses

Sleeping

tclopess commited on Jan 13, 2024

Commit

e1bbc3a

verified ·

1 Parent(s): 9aa9d1d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import onnxruntime as rt
 from transformers import AutoTokenizer
 import torch, json
-tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
 with open("genre_types_encoded.json", "r") as fp:
   encode_genre_types = json.load(fp)

 from transformers import AutoTokenizer
 import torch, json
+tokenizer = AutoTokenizer.from_pretrained("neuralmind/bert-large-portuguese-cased")
 with open("genre_types_encoded.json", "r") as fp:
   encode_genre_types = json.load(fp)