Views
No views yet
1from transformers import EncoderDecoderModel, BertTokenizer
2
3# Load the model and tokenizer
4model = EncoderDecoderModel.from_pretrained("mwz/UrduParaphraseBERT")
5tokenizer = BertTokenizer.from_pretrained("mwz/UrduParaphraseBERT")
6
7def paraphrase_urdu_sentence(sentence):
8 input_ids = tokenizer.encode(sentence, padding="longest", truncation=True, max_length=512, return_tensors="pt")
9 generated_ids = model.generate(input_ids=input_ids, max_length=128, num_beams=4, no_repeat_ngram_size=2)
10
11 paraphrase = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
12 return paraphrase
13sentence = "ایک مثالی روشنی کا مشہور نقطہ آبادی چھوٹی چھوٹی سڑکوں میں اپنے آپ کو خوشگوار کرسکتی ہے۔"
14paraphrased_sentence = paraphrase_urdu_sentence(sentence)
15print(paraphrased_sentence)