Update README.md

README.md

@@ -24,9 +24,13 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 max_length = 100
 src_lang = 'eng_Latn'
 tgt_lang = 'deu_Latn'
-context_text = 'This is an optional context sentence.'  # use '' empty string if not context should be used
+context_text = 'This is an optional context sentence.'
 sentence_text = 'Text to be translated.'
+
+# if the context is provided use the following:
 input_text = f'{context_text} {tokenizer.sep_token} {sentence_text}'
+# if no context is provided use the following:
+# input_text = sentence_text
 
 tokenizer.src_lang = src_lang
 inputs = tokenizer(input_text, return_tensors='pt').to(model.device)
@@ -46,9 +50,14 @@ model_name = 'voxreality/src_ctx_aware_nllb_1.3B'
 translation_pipeline = pipeline("translation", model=model_name)
 src_lang = 'eng_Latn'
 tgt_lang = 'deu_Latn'
-context_text = 'This is an optional context sentence.'  # use '' empty string if not context should be used
+context_text = 'This is an optional context sentence.'
 sentence_text = 'Text to be translated.'
-input_texts = [f'{context_text} {translation_pipeline.tokenizer.sep_token} {sentence_text}']
+
+# if the context is provided use the following:
+input_texts = [f'{context_text} {tokenizer.sep_token} {sentence_text}']
+# if no context is provided use the following:
+# input_texts = [sentence_text]
+
 
 pipeline_output = translation_pipeline(input_texts, src_lang=src_lang, tgt_lang=tgt_lang)