Var3n commited on
Commit
624f294
1 Parent(s): 3b50a06

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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(input, 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)
 
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)