Views
No views yet
1from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
2
3model = AutoModelForSeq2SeqLM.from_pretrained("Hnabil/t5-address-standardizer")
4tokenizer = AutoTokenizer.from_pretrained("Hnabil/t5-address-standardizer")
5
6inputs = tokenizer(
7 "220, soyth rhodeisland aveune, mason city, iowa, 50401, us",
8 return_tensors="pt"
9)
10outputs = model.generate(**inputs, max_length=100)
11print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
12
13# ['220, s rhode island ave, mason city, ia, 50401, us']