Edit-Banana/pyproject.toml at 1aabf62f20f2a2e879ddefc8fbcf6d02719a1b06 · BIT-DataLab/Edit-Banana · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
[project]
name = "edit-banana"
version = "0.1.0"
description = "Universal Content Re-Editor: Make the Uneditable, Editable"
readme = "README.md"
license = "Apache-2.0"
requires-python = ">=3.10"
dependencies = [
    "pyyaml",
    "opencv-python-headless",
    "numpy",
    "Pillow",
    "scikit-image",
    "requests",
    "fastapi",
    "uvicorn[standard]",
    "pytesseract",
]

[project.optional-dependencies]
paddleocr = [
    "paddlepaddle==3.2.2",
    "paddleocr",
]
formula = [
    "pix2text",
    "onnxruntime",
]
rmbg = [
    "onnxruntime",
]
torch = [
    "torch>=2.0",
    "torchvision",
]
sam3 = [
    "edit-banana[torch]",
    "sam3",
    "einops",
    "pycocotools",
    "psutil",
]

[project.scripts]
edit-banana = "main:main"

[tool.setuptools.packages.find]
include = ["modules*"]

[tool.uv]
dev-dependencies = []

[tool.uv.sources]
sam3 = { path = "sam3_src", editable = true }
# On macOS Apple Silicon, use default PyPI torch (includes MPS support).
# For Linux+CUDA, override via:
#   torch = { index = "pytorch-cu121" }
#   torchvision = { index = "pytorch-cu121" }