mirror of
https://github.com/yihong0618/bilingual_book_maker.git
synced 2025-06-06 11:35:49 +00:00
cd56b1e0ab
* feat: add gpt4 support * Update prompt_template_sample.json * fix: cleaned up formatting for quotes * feature: added context functionality (--use_context) for GPT4 model, which accumulates a running paragraph giving historical context to the current passage * fix: propagated context_flag argument to txt_loader and srt_loader * Updated Readme to include GPT4 parameters * Removed debug output * fix: lint --------- Co-authored-by: yihong <zouzou0208@gmail.com>
19 lines
601 B
Python
19 lines
601 B
Python
from book_maker.translator.caiyun_translator import Caiyun
|
|
from book_maker.translator.chatgptapi_translator import ChatGPTAPI
|
|
from book_maker.translator.deepl_translator import DeepL
|
|
from book_maker.translator.google_translator import Google
|
|
from book_maker.translator.gpt3_translator import GPT3
|
|
from book_maker.translator.gpt4_translator import GPT4
|
|
from book_maker.translator.claude_translator import Claude
|
|
|
|
MODEL_DICT = {
|
|
"chatgptapi": ChatGPTAPI,
|
|
"gpt3": GPT3,
|
|
"google": Google,
|
|
"caiyun": Caiyun,
|
|
"deepl": DeepL,
|
|
"gpt4": GPT4,
|
|
"claude": Claude,
|
|
# add more here
|
|
}
|