Spaces:

keras-io
/

text-generation

Build error

Jezia commited on Feb 12, 2022

Commit

fffdd3d

1 Parent(s): 51c97a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from huggingface_hub import from_pretrained_keras
-tokenizer = AutoTokenizer.from_pretrained("aditi2222/automatic_title_generation")
 model = from_pretrained_keras("keras-io/text-generation-miniature-gpt")
 def tokenize_data(text):

 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 from huggingface_hub import from_pretrained_keras
+tokenizer = AutoTokenizer.from_pretrained("bert-base-cased")
 model = from_pretrained_keras("keras-io/text-generation-miniature-gpt")
 def tokenize_data(text):