Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3# Load model and tokenizer
4model = AutoModelForSeq2SeqLM.from_pretrained("Paulwalker4884/facebook-persian")
5tokenizer = AutoTokenizer.from_pretrained("Paulwalker4884/facebook-persian")
6
7# Translation function
8def translate_persian_to_english(persian_text):
9 # Set source language to Persian
10 tokenizer.src_lang = "pes_Arab"
11
12 # Encode input text
13 inputs = tokenizer(persian_text, return_tensors="pt", padding=True, truncation=True)
14
15 # Generate translation
16 with torch.no_grad():
17 generated_tokens = model.generate(
18 **inputs,
19 forced_bos_token_id=tokenizer.lang_code_to_id["eng_Latn"],
20 max_length=256,
21 num_beams=4,
22 early_stopping=True
23 )
24
25 # Decode and return result
26 translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
27 return translation
28
29# Example usage
30persian_text = "سلام دنیا، چطور هستید؟"
31english_translation = translate_persian_to_english(persian_text)
32print(f"Persian: {persian_text}")
33print(f"English: {english_translation}")1from transformers import pipeline
2
3translator = pipeline("translation", model="Paulwalker4884/facebook-persian")
4result = translator("سلام دنیا", src_lang="pes_Arab", tgt_lang="eng_Latn")
5print(result[0]['translation_text'])| Persian (فارسی) | English |
|---|---|
| سلام دنیا | Hello world |
| چطور هستید؟ | How are you? |
| من دانشجوی دانشگاه هستم | I am a university student |
| امروز هوا خیلی خوب است | The weather is very nice today |
pip install transformers torch1@misc{persian-english-translator,
2 title={Persian to English Translation Model},
3 author={Yasin},
4 year={2024},
5 publisher={Hugging Face},
6 url={https://huggingface.co/Paulwalker4884/facebook-persian}
7}