Spaces:

JeCabrera
/

Michau-t5-base-en-generate-headline

Runtime error

JeCabrera commited on Feb 23

Commit

cb6ae08

verified ·

1 Parent(s): 8e7b067

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from transformers import T5ForConditionalGeneration,T5Tokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = T5ForConditionalGeneration.from_pretrained("Michau/t5-base-en-generate-headline")
-tokenizer = T5Tokenizer.from_pretrained("Michau/t5-base-en-generate-headline")
 model = model.to(device)
 article = '''

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = T5ForConditionalGeneration.from_pretrained("Michau/t5-base-en-generate-headline")
+tokenizer = T5Tokenizer.from_pretrained("Michau/t5-base-en-generate-headline", legacy=False)
 model = model.to(device)
 article = '''