fix: ci and readme problem

This commit is contained in:
yihong0618 2023-05-08 10:19:30 +08:00
parent 8f99d0146d
commit 7322d4a197
4 changed files with 7 additions and 20 deletions

View File

@ -98,7 +98,7 @@ python3 make_book.py --book_name test_books/the_little_prince.txt --test --batch
# (the api currently only support: simplified chinese <-> english, simplified chinese <-> japanese) # (the api currently only support: simplified chinese <-> english, simplified chinese <-> japanese)
# the official Caiyun has provided a test token (3975l6lr5pcbvidl6jl2) # the official Caiyun has provided a test token (3975l6lr5pcbvidl6jl2)
# you can apply your own token by following this tutorial(https://bobtranslate.com/service/translate/caiyun.html) # you can apply your own token by following this tutorial(https://bobtranslate.com/service/translate/caiyun.html)
python3 make_book.py --model caiyun --openai_key 3975l6lr5pcbvidl6jl2 --book_name test_books/animal_farm.epub python3 make_book.py --model caiyun --caiyun_key 3975l6lr5pcbvidl6jl2 --book_name test_books/animal_farm.epub
# Set env BBM_CAIYUN_API_KEY to ignore option --openai_key # Set env BBM_CAIYUN_API_KEY to ignore option --openai_key

View File

@ -1,8 +1 @@
bs4 -e .
openai
requests
ebooklib
rich
tqdm
tiktoken
revChatGPT

View File

@ -2,26 +2,20 @@
from setuptools import find_packages, setup from setuptools import find_packages, setup
def get_required_packges(): packages = ["bs4", "openai", "requests", "ebooklib", "rich", "tqdm", "tiktoken"]
packages = []
with open("requirements.txt") as filep:
for line in filep:
packages.append(line.rstrip())
return packages
setup( setup(
name="bbook_maker", name="bbook_maker",
description="The bilingual_book_maker is an AI translation tool that uses ChatGPT to assist users in creating multi-language versions of epub/txt files and books.", description="The bilingual_book_maker is an AI translation tool that uses ChatGPT to assist users in creating multi-language versions of epub/txt files and books.",
version="0.3.0", version="0.4.0",
license="MIT", license="MIT",
author="yihong0618", author="yihong0618",
author_email="zouzou0208@gmail.com", author_email="zouzou0208@gmail.com",
packages=find_packages(), packages=find_packages(),
url="https://github.com/yihong0618/bilingual_book_maker", url="https://github.com/yihong0618/bilingual_book_maker",
python_requires=">=3.7", python_requires=">=3.7",
install_requires=get_required_packges(), install_requires=packages,
classifiers=[ classifiers=[
"Programming Language :: Python :: 3", "Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License", "License :: OSI Approved :: MIT License",

View File

@ -115,7 +115,7 @@ def test_caiyun_translate_txt(test_book_dir, tmpdir):
os.path.join(tmpdir, "the_little_prince.txt"), os.path.join(tmpdir, "the_little_prince.txt"),
"--test", "--test",
"--batch_size", "--batch_size",
"30", "10",
"--test_num", "--test_num",
"100", "100",
"--model", "--model",
@ -179,7 +179,7 @@ def test_deepl_translate_srt(test_book_dir, tmpdir):
"--batch_size", "--batch_size",
"30", "30",
"--test_num", "--test_num",
"20", "2",
"--model", "--model",
"deepl", "deepl",
], ],