mirrors/LibreTranslate
1
1
Fork 1
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2024-11-21 23:41:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

Fix alternatives unique comparison

Browse source
This commit is contained in:
Piero Toffanin 2024-06-03 13:11:42 -04:00
parent cb6b324478
commit df3680c280
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
libretranslate/app.py
View file

@ -148,8 +148,8 @@ def get_routes_limits(args, api_keys_db):
return res return res
def unique_list(seq): def unique_list(seq, extra):
seen = set() seen = set({extra})
seen_add = seen.add seen_add = seen.add
return [x for x in seq if not (x in seen or seen_add(x))] return [x for x in seq if not (x in seen or seen_add(x))]
@ -661,7 +661,7 @@ def create_app(args):
else: else:
hypotheses = translator.hypotheses(text, num_alternatives + 1) hypotheses = translator.hypotheses(text, num_alternatives + 1)
translated_text = unescape(improve_translation_formatting(text, hypotheses[0].value)) translated_text = unescape(improve_translation_formatting(text, hypotheses[0].value))
alternatives = unique_list([unescape(improve_translation_formatting(text, hypotheses[i].value)) for i in range(1, len(hypotheses))]) alternatives = unique_list([unescape(improve_translation_formatting(text, hypotheses[i].value)) for i in range(1, len(hypotheses))], translated_text)
batch_results.append(translated_text) batch_results.append(translated_text)
batch_alternatives.append(alternatives) batch_alternatives.append(alternatives)
@ -685,7 +685,7 @@ def create_app(args):
else: else:
hypotheses = translator.hypotheses(q, num_alternatives + 1) hypotheses = translator.hypotheses(q, num_alternatives + 1)
translated_text = unescape(improve_translation_formatting(q, hypotheses[0].value)) translated_text = unescape(improve_translation_formatting(q, hypotheses[0].value))
alternatives = unique_list([unescape(improve_translation_formatting(q, hypotheses[i].value)) for i in range(1, len(hypotheses))]) alternatives = unique_list([unescape(improve_translation_formatting(q, hypotheses[i].value)) for i in range(1, len(hypotheses))], translated_text)
result = {"translatedText": translated_text} result = {"translatedText": translated_text}