Views
No views yet
1
2from transformers import ByT5Tokenizer, T5ForConditionalGeneration
3
4# Initialize the tokenizer and model
5tokenizer = ByT5Tokenizer.from_pretrained('saadamin2k13/italian_augmented_semantic_parser', max_length=512)
6
7model = T5ForConditionalGeneration.from_pretrained('saadamin2k13/italian_augmented_semantic_parser')
8
9# Example sentence
10example = "questa macchina è nera."
11
12# Tokenize and prepare the input
13x = tokenizer(example, return_tensors='pt', padding=True, truncation=True, max_length=512)['input_ids']
14
15# Generate output
16output = model.generate(x)
17
18# Decode and print the output text
19pred_text = tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=False)
20print(pred_text)
21