Fralet commited on
Commit
2d7e9c0
1 Parent(s): ec5e298

Create app2.py

Browse files
Files changed (1) hide show
  1. app2.py +38 -0
app2.py ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import logging
2
+ from transformers import MarianMTModel, MarianTokenizer
3
+ import sentencepiece
4
+
5
+ # Setup logging
6
+ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
7
+
8
+ # Initialize logger
9
+ logger = logging.getLogger(__name__)
10
+
11
+ # Define a function to translate text from Russian to English
12
+ def translate_to_english(text):
13
+ logger.info("Loading model and tokenizer")
14
+ model_name = 'Helsinki-NLP/opus-mt-ru-en'
15
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
16
+ model = MarianMTModel.from_pretrained(model_name)
17
+
18
+ logger.info("Tokenizing text")
19
+ tokenized_text = tokenizer(text, return_tensors='pt', padding=True, truncation=True)
20
+
21
+ logger.info("Performing translation")
22
+ translated_tokens = model.generate(**tokenized_text)
23
+
24
+ logger.info("Decoding translation")
25
+ translated_text = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
26
+ return translated_text
27
+
28
+ # Sample Russian text
29
+ russian_text = "16 стран-участниц «мирного саммита» в Швейцарии отказались подписывать коммюнике. Изначально документ подписали 80 из 92 приглашённых государств. В частности, поддержать текст отказались Армения, Бахрейн, Бразилия, Индия, Индонезия, Ливия, Мексика, ОАЭ, Саудовская Аравия, Таиланд, ЮАР. Святой престол Папы Римского также сначала отказался подписывать коммюнике, но вскоре передумал. Свои подписи под итоговой декларацией вскоре отозвали Ирак и Иордания. Теперь из списка также исчезла Руанда. Примечательно, что в день саммита министр иностранных дел Украины Кулеба лично встретился с делегацией из Кигали и даже отчитался об этом. Представители государства не сообщили, почему решили отозвать свою подпись. Украинская сторона также не прокомментировала решение. Ранее выяснилось, что суть подписанного на саммите коммюнике свелась к предупреждению о недопустимости применения ядерного оружия и предложениям по обмену пленными."
30
+
31
+ logger.info("Starting translation")
32
+ # Translate the text
33
+ english_text = translate_to_english(russian_text)
34
+
35
+ logger.info("Translation complete")
36
+ # Print the translated text
37
+ print(f"Original Russian text: {russian_text}")
38
+ print(f"Translated English text: {english_text}")