Update README.md
Browse files
README.md
CHANGED
@@ -30,7 +30,7 @@ example_sentence = "Anvpreiſungq. Haupidepot für Wien: In der Stadt, obere Br
|
|
30 |
tokenizer = AutoTokenizer.from_pretrained("Var3n/hmByT5_anno")
|
31 |
model = AutoModelForSeq2SeqLM.from_pretrained("Var3n/hmByT5_anno")
|
32 |
|
33 |
-
input = tokenizer(
|
34 |
output = model.generate(input, max_new_tokens=len(input[0]), num_beams=4, do_sample=True)
|
35 |
|
36 |
text = tokenizer.decode(output[0], skip_special_tokens=True)
|
|
|
30 |
tokenizer = AutoTokenizer.from_pretrained("Var3n/hmByT5_anno")
|
31 |
model = AutoModelForSeq2SeqLM.from_pretrained("Var3n/hmByT5_anno")
|
32 |
|
33 |
+
input = tokenizer(example_sentence, return_tensors="pt").input_ids
|
34 |
output = model.generate(input, max_new_tokens=len(input[0]), num_beams=4, do_sample=True)
|
35 |
|
36 |
text = tokenizer.decode(output[0], skip_special_tokens=True)
|