Views
No views yet
1
2from transformers import ByT5Tokenizer, T5ForConditionalGeneration
3
4# Initialize the tokenizer and model
5tokenizer = ByT5Tokenizer.from_pretrained('saadamin2k13/urdu_semantic_parsing', max_length=512)
6
7model = T5ForConditionalGeneration.from_pretrained('saadamin2k13/urdu_semantic_parsing')
8
9# Example sentence
10example = "یہ کار کالی ہے۔"
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