Spaces:

waddaheaven
/

multilabel-movie-genre-classifier

Running

waddaheaven commited on Jul 13

Commit

50c8092

•

1 Parent(s): 220e363

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 import torch
 import json
+tokenizer = AutoTokenizer.from_pretrained("distilroberta-base")
 with open("genre_types_encoded.json", "r") as fp:
     encode_genre_types = json.load(fp)