Commit c349c47b authored by dmMaze's avatar dmMaze
Browse files

set lama_mape as default inpainter

parent fa8f579e

ballontranslator/dl/translators/__init__.py

+4 −2
Original line number Diff line number Diff line
@@ -5,7 +5,7 @@ import functools 
import json

from .exceptions import InvalidSourceOrTargetLanguage, TranslatorSetupFailure, MissingTranslatorParams, TranslatorNotValid

from ..textdetector.textblock import TextBlock

from ..moduleparamparser import ModuleParamParser, DEFAULT_DEVICE

from ..moduleparamparser import ModuleParamParser

from utils.registry import Registry

from utils.io_utils import text_is_empty

import deepl
@@ -159,6 +159,7 @@ class TranslatorBase(ModuleParamParser): 
                return True

        return False





@register_translator('google')

class GoogleTranslator(TranslatorBase):
@@ -261,7 +262,6 @@ class PapagoTranslator(TranslatorBase): 
        return translations

        





@register_translator('caiyun')

class CaiyunTranslator(TranslatorBase):
@@ -300,6 +300,7 @@ class CaiyunTranslator(TranslatorBase): 


        return translations





@register_translator('Deepl')

class DeeplTranslator(TranslatorBase):
@@ -344,6 +345,7 @@ class DeeplTranslator(TranslatorBase): 
        result = translator.translate_text(text, source_lang=source, target_lang=target)

        return [i.text for i in result]

    



# # "dummy translator" is the name showed in the app

# @register_translator('dummy translator')

# class DummyTranslator(TranslatorBase):

ballontranslator/ui/misc.py

+1 −1
Original line number Diff line number Diff line
@@ -367,7 +367,7 @@ class DLModuleConfig: 
    def __init__(self, 

                 textdetector: str = 'ctd',

                 ocr = "mit32px",

                 inpainter: str = 'aot',

                 inpainter: str = 'lama_mpe',

                 translator = "google",

                 enable_ocr = True,

                 enable_translate = True,

ballontranslator/ui/scenetext_manager.py

+2 −1
Original line number Diff line number Diff line
@@ -609,6 +609,7 @@ class SceneTextManager(QObject): 
        

        blkitem.setPlainText(new_text)

        blkitem.setRect(xywh)

        if len(self.pairwidget_list) > blkitem.idx:

            self.pairwidget_list[blkitem.idx].e_trans.setPlainText(new_text)

        if restore_charfmts:

            self.restore_charfmts(blkitem, text, new_text, char_fmts)