Skip to content

Commit 54177fa

Browse files
Update app.py
1 parent 4988f52 commit 54177fa

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

app.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -357,15 +357,14 @@ def __init__(self, Video_path, source_language, target_language,
357357
'None': None}
358358

359359

360-
361360
if self.source_language == 'tr':
362361
model_name = f"Helsinki-NLP/opus-mt-trk-{self.target_language}"
363362
elif self.target_language == 'tr':
364363
model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-trk"
365364
else:
366365
model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-{self.target_language}"
367366

368-
if self.source_language == 'zh-cn':
367+
if self.source_language == 'zh-cn':
369368
model_name = f"Helsinki-NLP/opus-mt-zh-{self.target_language}"
370369
elif self.target_language == 'zh-cn':
371370
model_name = f"Helsinki-NLP/opus-mt-{self.source_language}-zh"
@@ -376,8 +375,6 @@ def __init__(self, Video_path, source_language, target_language,
376375
model = MarianMTModel.from_pretrained(model_name).to(device)
377376

378377
if not self.Context_translation:
379-
380-
381378

382379
# Function to translate text
383380
def translate(sentence, model, tokenizer):

0 commit comments

Comments
 (0)