Spaces:
Runtime error
Runtime error
Iliya Zhechev
commited on
Commit
•
21fcd0a
1
Parent(s):
a6c9838
simple backed init
Browse files- .vscode/settings.json +3 -0
- poetry.lock +690 -0
- pyproject.toml +20 -0
- tests/__init__.py +0 -0
- tests/test_text_brush.py +5 -0
- text_brush/.gitignore +1 -0
- text_brush/__init__.py +1 -0
- text_brush/app.py +64 -0
.vscode/settings.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"python.formatting.provider": "black"
|
3 |
+
}
|
poetry.lock
ADDED
@@ -0,0 +1,690 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[[package]]
|
2 |
+
name = "atomicwrites"
|
3 |
+
version = "1.4.1"
|
4 |
+
description = "Atomic file writes."
|
5 |
+
category = "dev"
|
6 |
+
optional = false
|
7 |
+
python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
|
8 |
+
|
9 |
+
[[package]]
|
10 |
+
name = "attrs"
|
11 |
+
version = "22.1.0"
|
12 |
+
description = "Classes Without Boilerplate"
|
13 |
+
category = "dev"
|
14 |
+
optional = false
|
15 |
+
python-versions = ">=3.5"
|
16 |
+
|
17 |
+
[package.extras]
|
18 |
+
dev = ["coverage[toml] (>=5.0.2)", "hypothesis", "pympler", "pytest (>=4.3.0)", "mypy (>=0.900,!=0.940)", "pytest-mypy-plugins", "zope.interface", "furo", "sphinx", "sphinx-notfound-page", "pre-commit", "cloudpickle"]
|
19 |
+
docs = ["furo", "sphinx", "zope.interface", "sphinx-notfound-page"]
|
20 |
+
tests = ["coverage[toml] (>=5.0.2)", "hypothesis", "pympler", "pytest (>=4.3.0)", "mypy (>=0.900,!=0.940)", "pytest-mypy-plugins", "zope.interface", "cloudpickle"]
|
21 |
+
tests_no_zope = ["coverage[toml] (>=5.0.2)", "hypothesis", "pympler", "pytest (>=4.3.0)", "mypy (>=0.900,!=0.940)", "pytest-mypy-plugins", "cloudpickle"]
|
22 |
+
|
23 |
+
[[package]]
|
24 |
+
name = "black"
|
25 |
+
version = "22.10.0"
|
26 |
+
description = "The uncompromising code formatter."
|
27 |
+
category = "dev"
|
28 |
+
optional = false
|
29 |
+
python-versions = ">=3.7"
|
30 |
+
|
31 |
+
[package.dependencies]
|
32 |
+
click = ">=8.0.0"
|
33 |
+
mypy-extensions = ">=0.4.3"
|
34 |
+
pathspec = ">=0.9.0"
|
35 |
+
platformdirs = ">=2"
|
36 |
+
tomli = {version = ">=1.1.0", markers = "python_full_version < \"3.11.0a7\""}
|
37 |
+
typing-extensions = {version = ">=3.10.0.0", markers = "python_version < \"3.10\""}
|
38 |
+
|
39 |
+
[package.extras]
|
40 |
+
colorama = ["colorama (>=0.4.3)"]
|
41 |
+
d = ["aiohttp (>=3.7.4)"]
|
42 |
+
jupyter = ["ipython (>=7.8.0)", "tokenize-rt (>=3.2.0)"]
|
43 |
+
uvloop = ["uvloop (>=0.15.2)"]
|
44 |
+
|
45 |
+
[[package]]
|
46 |
+
name = "certifi"
|
47 |
+
version = "2022.9.24"
|
48 |
+
description = "Python package for providing Mozilla's CA Bundle."
|
49 |
+
category = "main"
|
50 |
+
optional = false
|
51 |
+
python-versions = ">=3.6"
|
52 |
+
|
53 |
+
[[package]]
|
54 |
+
name = "charset-normalizer"
|
55 |
+
version = "2.1.1"
|
56 |
+
description = "The Real First Universal Charset Detector. Open, modern and actively maintained alternative to Chardet."
|
57 |
+
category = "main"
|
58 |
+
optional = false
|
59 |
+
python-versions = ">=3.6.0"
|
60 |
+
|
61 |
+
[package.extras]
|
62 |
+
unicode_backport = ["unicodedata2"]
|
63 |
+
|
64 |
+
[[package]]
|
65 |
+
name = "click"
|
66 |
+
version = "8.1.3"
|
67 |
+
description = "Composable command line interface toolkit"
|
68 |
+
category = "main"
|
69 |
+
optional = false
|
70 |
+
python-versions = ">=3.7"
|
71 |
+
|
72 |
+
[package.dependencies]
|
73 |
+
colorama = {version = "*", markers = "platform_system == \"Windows\""}
|
74 |
+
|
75 |
+
[[package]]
|
76 |
+
name = "colorama"
|
77 |
+
version = "0.4.5"
|
78 |
+
description = "Cross-platform colored terminal text."
|
79 |
+
category = "main"
|
80 |
+
optional = false
|
81 |
+
python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
|
82 |
+
|
83 |
+
[[package]]
|
84 |
+
name = "diffusers"
|
85 |
+
version = "0.4.1"
|
86 |
+
description = "Diffusers"
|
87 |
+
category = "main"
|
88 |
+
optional = false
|
89 |
+
python-versions = ">=3.7.0"
|
90 |
+
|
91 |
+
[package.dependencies]
|
92 |
+
filelock = "*"
|
93 |
+
huggingface-hub = ">=0.10.0"
|
94 |
+
importlib-metadata = "*"
|
95 |
+
numpy = "*"
|
96 |
+
Pillow = "<10.0"
|
97 |
+
regex = "!=2019.12.17"
|
98 |
+
requests = "*"
|
99 |
+
|
100 |
+
[package.extras]
|
101 |
+
training = ["modelcards (>=0.1.4)", "tensorboard", "datasets", "accelerate (>=0.11.0)"]
|
102 |
+
torch = ["torch (>=1.4)"]
|
103 |
+
test = ["transformers (>=4.21.0)", "torchvision", "scipy", "pytest-xdist", "pytest-timeout", "pytest", "onnxruntime", "datasets", "accelerate (>=0.11.0)"]
|
104 |
+
quality = ["hf-doc-builder (>=0.3.0)", "flake8 (>=3.8.3)", "isort (>=5.5.4)", "black (==22.8)"]
|
105 |
+
flax = ["flax (>=0.4.1)", "jaxlib (>=0.1.65,<=0.3.6)", "jax (>=0.2.8,!=0.3.2,<=0.3.6)"]
|
106 |
+
docs = ["hf-doc-builder (>=0.3.0)"]
|
107 |
+
dev = ["flax (>=0.4.1)", "jaxlib (>=0.1.65,<=0.3.6)", "jax (>=0.2.8,!=0.3.2,<=0.3.6)", "torch (>=1.4)", "modelcards (>=0.1.4)", "tensorboard", "transformers (>=4.21.0)", "torchvision", "scipy", "pytest-xdist", "pytest-timeout", "pytest", "onnxruntime", "datasets", "accelerate (>=0.11.0)", "hf-doc-builder (>=0.3.0)", "flake8 (>=3.8.3)", "isort (>=5.5.4)", "black (==22.8)"]
|
108 |
+
|
109 |
+
[[package]]
|
110 |
+
name = "filelock"
|
111 |
+
version = "3.8.0"
|
112 |
+
description = "A platform independent file lock."
|
113 |
+
category = "main"
|
114 |
+
optional = false
|
115 |
+
python-versions = ">=3.7"
|
116 |
+
|
117 |
+
[package.extras]
|
118 |
+
docs = ["furo (>=2022.6.21)", "sphinx (>=5.1.1)", "sphinx-autodoc-typehints (>=1.19.1)"]
|
119 |
+
testing = ["covdefaults (>=2.2)", "coverage (>=6.4.2)", "pytest (>=7.1.2)", "pytest-cov (>=3)", "pytest-timeout (>=2.1)"]
|
120 |
+
|
121 |
+
[[package]]
|
122 |
+
name = "flask"
|
123 |
+
version = "2.2.2"
|
124 |
+
description = "A simple framework for building complex web applications."
|
125 |
+
category = "main"
|
126 |
+
optional = false
|
127 |
+
python-versions = ">=3.7"
|
128 |
+
|
129 |
+
[package.dependencies]
|
130 |
+
click = ">=8.0"
|
131 |
+
importlib-metadata = {version = ">=3.6.0", markers = "python_version < \"3.10\""}
|
132 |
+
itsdangerous = ">=2.0"
|
133 |
+
Jinja2 = ">=3.0"
|
134 |
+
Werkzeug = ">=2.2.2"
|
135 |
+
|
136 |
+
[package.extras]
|
137 |
+
async = ["asgiref (>=3.2)"]
|
138 |
+
dotenv = ["python-dotenv"]
|
139 |
+
|
140 |
+
[[package]]
|
141 |
+
name = "huggingface-hub"
|
142 |
+
version = "0.10.0"
|
143 |
+
description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
|
144 |
+
category = "main"
|
145 |
+
optional = false
|
146 |
+
python-versions = ">=3.7.0"
|
147 |
+
|
148 |
+
[package.dependencies]
|
149 |
+
filelock = "*"
|
150 |
+
packaging = ">=20.9"
|
151 |
+
pyyaml = ">=5.1"
|
152 |
+
requests = "*"
|
153 |
+
tqdm = "*"
|
154 |
+
typing-extensions = ">=3.7.4.3"
|
155 |
+
|
156 |
+
[package.extras]
|
157 |
+
torch = ["torch"]
|
158 |
+
testing = ["soundfile", "pytest-cov", "pytest", "jinja2", "jedi", "isort (>=5.5.4)", "InquirerPy (==0.3.4)"]
|
159 |
+
tensorflow = ["graphviz", "pydot", "tensorflow"]
|
160 |
+
quality = ["mypy", "isort (>=5.5.4)", "flake8-bugbear", "flake8 (>=3.8.3)", "black (==22.3)"]
|
161 |
+
fastai = ["fastcore (>=1.3.27)", "fastai (>=2.4)", "toml"]
|
162 |
+
dev = ["mypy", "flake8-bugbear", "flake8 (>=3.8.3)", "black (==22.3)", "soundfile", "pytest-cov", "pytest", "jinja2", "jedi", "isort (>=5.5.4)", "InquirerPy (==0.3.4)"]
|
163 |
+
cli = ["InquirerPy (==0.3.4)"]
|
164 |
+
all = ["mypy", "flake8-bugbear", "flake8 (>=3.8.3)", "black (==22.3)", "soundfile", "pytest-cov", "pytest", "jinja2", "jedi", "isort (>=5.5.4)", "InquirerPy (==0.3.4)"]
|
165 |
+
|
166 |
+
[[package]]
|
167 |
+
name = "idna"
|
168 |
+
version = "3.4"
|
169 |
+
description = "Internationalized Domain Names in Applications (IDNA)"
|
170 |
+
category = "main"
|
171 |
+
optional = false
|
172 |
+
python-versions = ">=3.5"
|
173 |
+
|
174 |
+
[[package]]
|
175 |
+
name = "importlib-metadata"
|
176 |
+
version = "5.0.0"
|
177 |
+
description = "Read metadata from Python packages"
|
178 |
+
category = "main"
|
179 |
+
optional = false
|
180 |
+
python-versions = ">=3.7"
|
181 |
+
|
182 |
+
[package.dependencies]
|
183 |
+
zipp = ">=0.5"
|
184 |
+
|
185 |
+
[package.extras]
|
186 |
+
docs = ["sphinx (>=3.5)", "jaraco.packaging (>=9)", "rst.linker (>=1.9)", "furo", "jaraco.tidelift (>=1.4)"]
|
187 |
+
perf = ["ipython"]
|
188 |
+
testing = ["pytest (>=6)", "pytest-checkdocs (>=2.4)", "pytest-flake8", "flake8 (<5)", "pytest-cov", "pytest-enabler (>=1.3)", "packaging", "pyfakefs", "flufl.flake8", "pytest-perf (>=0.9.2)", "pytest-black (>=0.3.7)", "pytest-mypy (>=0.9.1)", "importlib-resources (>=1.3)"]
|
189 |
+
|
190 |
+
[[package]]
|
191 |
+
name = "itsdangerous"
|
192 |
+
version = "2.1.2"
|
193 |
+
description = "Safely pass data to untrusted environments and back."
|
194 |
+
category = "main"
|
195 |
+
optional = false
|
196 |
+
python-versions = ">=3.7"
|
197 |
+
|
198 |
+
[[package]]
|
199 |
+
name = "jinja2"
|
200 |
+
version = "3.1.2"
|
201 |
+
description = "A very fast and expressive template engine."
|
202 |
+
category = "main"
|
203 |
+
optional = false
|
204 |
+
python-versions = ">=3.7"
|
205 |
+
|
206 |
+
[package.dependencies]
|
207 |
+
MarkupSafe = ">=2.0"
|
208 |
+
|
209 |
+
[package.extras]
|
210 |
+
i18n = ["Babel (>=2.7)"]
|
211 |
+
|
212 |
+
[[package]]
|
213 |
+
name = "markupsafe"
|
214 |
+
version = "2.1.1"
|
215 |
+
description = "Safely add untrusted strings to HTML/XML markup."
|
216 |
+
category = "main"
|
217 |
+
optional = false
|
218 |
+
python-versions = ">=3.7"
|
219 |
+
|
220 |
+
[[package]]
|
221 |
+
name = "more-itertools"
|
222 |
+
version = "8.14.0"
|
223 |
+
description = "More routines for operating on iterables, beyond itertools"
|
224 |
+
category = "dev"
|
225 |
+
optional = false
|
226 |
+
python-versions = ">=3.5"
|
227 |
+
|
228 |
+
[[package]]
|
229 |
+
name = "mypy-extensions"
|
230 |
+
version = "0.4.3"
|
231 |
+
description = "Experimental type system extensions for programs checked with the mypy typechecker."
|
232 |
+
category = "dev"
|
233 |
+
optional = false
|
234 |
+
python-versions = "*"
|
235 |
+
|
236 |
+
[[package]]
|
237 |
+
name = "numpy"
|
238 |
+
version = "1.23.3"
|
239 |
+
description = "NumPy is the fundamental package for array computing with Python."
|
240 |
+
category = "main"
|
241 |
+
optional = false
|
242 |
+
python-versions = ">=3.8"
|
243 |
+
|
244 |
+
[[package]]
|
245 |
+
name = "packaging"
|
246 |
+
version = "21.3"
|
247 |
+
description = "Core utilities for Python packages"
|
248 |
+
category = "main"
|
249 |
+
optional = false
|
250 |
+
python-versions = ">=3.6"
|
251 |
+
|
252 |
+
[package.dependencies]
|
253 |
+
pyparsing = ">=2.0.2,<3.0.5 || >3.0.5"
|
254 |
+
|
255 |
+
[[package]]
|
256 |
+
name = "pathspec"
|
257 |
+
version = "0.10.1"
|
258 |
+
description = "Utility library for gitignore style pattern matching of file paths."
|
259 |
+
category = "dev"
|
260 |
+
optional = false
|
261 |
+
python-versions = ">=3.7"
|
262 |
+
|
263 |
+
[[package]]
|
264 |
+
name = "pillow"
|
265 |
+
version = "9.2.0"
|
266 |
+
description = "Python Imaging Library (Fork)"
|
267 |
+
category = "main"
|
268 |
+
optional = false
|
269 |
+
python-versions = ">=3.7"
|
270 |
+
|
271 |
+
[package.extras]
|
272 |
+
docs = ["furo", "olefile", "sphinx (>=2.4)", "sphinx-copybutton", "sphinx-issues (>=3.0.1)", "sphinx-removed-in", "sphinxext-opengraph"]
|
273 |
+
tests = ["check-manifest", "coverage", "defusedxml", "markdown2", "olefile", "packaging", "pyroma", "pytest", "pytest-cov", "pytest-timeout"]
|
274 |
+
|
275 |
+
[[package]]
|
276 |
+
name = "platformdirs"
|
277 |
+
version = "2.5.2"
|
278 |
+
description = "A small Python module for determining appropriate platform-specific dirs, e.g. a \"user data dir\"."
|
279 |
+
category = "dev"
|
280 |
+
optional = false
|
281 |
+
python-versions = ">=3.7"
|
282 |
+
|
283 |
+
[package.extras]
|
284 |
+
docs = ["furo (>=2021.7.5b38)", "proselint (>=0.10.2)", "sphinx-autodoc-typehints (>=1.12)", "sphinx (>=4)"]
|
285 |
+
test = ["appdirs (==1.4.4)", "pytest-cov (>=2.7)", "pytest-mock (>=3.6)", "pytest (>=6)"]
|
286 |
+
|
287 |
+
[[package]]
|
288 |
+
name = "pluggy"
|
289 |
+
version = "0.13.1"
|
290 |
+
description = "plugin and hook calling mechanisms for python"
|
291 |
+
category = "dev"
|
292 |
+
optional = false
|
293 |
+
python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
|
294 |
+
|
295 |
+
[package.extras]
|
296 |
+
dev = ["pre-commit", "tox"]
|
297 |
+
|
298 |
+
[[package]]
|
299 |
+
name = "py"
|
300 |
+
version = "1.11.0"
|
301 |
+
description = "library with cross-python path, ini-parsing, io, code, log facilities"
|
302 |
+
category = "dev"
|
303 |
+
optional = false
|
304 |
+
python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
|
305 |
+
|
306 |
+
[[package]]
|
307 |
+
name = "pyparsing"
|
308 |
+
version = "3.0.9"
|
309 |
+
description = "pyparsing module - Classes and methods to define and execute parsing grammars"
|
310 |
+
category = "main"
|
311 |
+
optional = false
|
312 |
+
python-versions = ">=3.6.8"
|
313 |
+
|
314 |
+
[package.extras]
|
315 |
+
diagrams = ["railroad-diagrams", "jinja2"]
|
316 |
+
|
317 |
+
[[package]]
|
318 |
+
name = "pytest"
|
319 |
+
version = "5.4.3"
|
320 |
+
description = "pytest: simple powerful testing with Python"
|
321 |
+
category = "dev"
|
322 |
+
optional = false
|
323 |
+
python-versions = ">=3.5"
|
324 |
+
|
325 |
+
[package.dependencies]
|
326 |
+
atomicwrites = {version = ">=1.0", markers = "sys_platform == \"win32\""}
|
327 |
+
attrs = ">=17.4.0"
|
328 |
+
colorama = {version = "*", markers = "sys_platform == \"win32\""}
|
329 |
+
more-itertools = ">=4.0.0"
|
330 |
+
packaging = "*"
|
331 |
+
pluggy = ">=0.12,<1.0"
|
332 |
+
py = ">=1.5.0"
|
333 |
+
wcwidth = "*"
|
334 |
+
|
335 |
+
[package.extras]
|
336 |
+
checkqa-mypy = ["mypy (==v0.761)"]
|
337 |
+
testing = ["argcomplete", "hypothesis (>=3.56)", "mock", "nose", "requests", "xmlschema"]
|
338 |
+
|
339 |
+
[[package]]
|
340 |
+
name = "pyyaml"
|
341 |
+
version = "6.0"
|
342 |
+
description = "YAML parser and emitter for Python"
|
343 |
+
category = "main"
|
344 |
+
optional = false
|
345 |
+
python-versions = ">=3.6"
|
346 |
+
|
347 |
+
[[package]]
|
348 |
+
name = "regex"
|
349 |
+
version = "2022.9.13"
|
350 |
+
description = "Alternative regular expression module, to replace re."
|
351 |
+
category = "main"
|
352 |
+
optional = false
|
353 |
+
python-versions = ">=3.6"
|
354 |
+
|
355 |
+
[[package]]
|
356 |
+
name = "requests"
|
357 |
+
version = "2.28.1"
|
358 |
+
description = "Python HTTP for Humans."
|
359 |
+
category = "main"
|
360 |
+
optional = false
|
361 |
+
python-versions = ">=3.7, <4"
|
362 |
+
|
363 |
+
[package.dependencies]
|
364 |
+
certifi = ">=2017.4.17"
|
365 |
+
charset-normalizer = ">=2,<3"
|
366 |
+
idna = ">=2.5,<4"
|
367 |
+
urllib3 = ">=1.21.1,<1.27"
|
368 |
+
|
369 |
+
[package.extras]
|
370 |
+
socks = ["PySocks (>=1.5.6,!=1.5.7)"]
|
371 |
+
use_chardet_on_py3 = ["chardet (>=3.0.2,<6)"]
|
372 |
+
|
373 |
+
[[package]]
|
374 |
+
name = "tokenizers"
|
375 |
+
version = "0.12.1"
|
376 |
+
description = "Fast and Customizable Tokenizers"
|
377 |
+
category = "main"
|
378 |
+
optional = false
|
379 |
+
python-versions = "*"
|
380 |
+
|
381 |
+
[package.extras]
|
382 |
+
docs = ["sphinx", "sphinx-rtd-theme", "setuptools-rust"]
|
383 |
+
testing = ["pytest", "requests", "numpy", "datasets"]
|
384 |
+
|
385 |
+
[[package]]
|
386 |
+
name = "tomli"
|
387 |
+
version = "2.0.1"
|
388 |
+
description = "A lil' TOML parser"
|
389 |
+
category = "dev"
|
390 |
+
optional = false
|
391 |
+
python-versions = ">=3.7"
|
392 |
+
|
393 |
+
[[package]]
|
394 |
+
name = "torch"
|
395 |
+
version = "1.12.1"
|
396 |
+
description = "Tensors and Dynamic neural networks in Python with strong GPU acceleration"
|
397 |
+
category = "main"
|
398 |
+
optional = false
|
399 |
+
python-versions = ">=3.7.0"
|
400 |
+
|
401 |
+
[package.dependencies]
|
402 |
+
typing-extensions = "*"
|
403 |
+
|
404 |
+
[[package]]
|
405 |
+
name = "tqdm"
|
406 |
+
version = "4.64.1"
|
407 |
+
description = "Fast, Extensible Progress Meter"
|
408 |
+
category = "main"
|
409 |
+
optional = false
|
410 |
+
python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,>=2.7"
|
411 |
+
|
412 |
+
[package.dependencies]
|
413 |
+
colorama = {version = "*", markers = "platform_system == \"Windows\""}
|
414 |
+
|
415 |
+
[package.extras]
|
416 |
+
dev = ["py-make (>=0.1.0)", "twine", "wheel"]
|
417 |
+
notebook = ["ipywidgets (>=6)"]
|
418 |
+
slack = ["slack-sdk"]
|
419 |
+
telegram = ["requests"]
|
420 |
+
|
421 |
+
[[package]]
|
422 |
+
name = "transformers"
|
423 |
+
version = "4.22.2"
|
424 |
+
description = "State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow"
|
425 |
+
category = "main"
|
426 |
+
optional = false
|
427 |
+
python-versions = ">=3.7.0"
|
428 |
+
|
429 |
+
[package.dependencies]
|
430 |
+
filelock = "*"
|
431 |
+
huggingface-hub = ">=0.9.0,<1.0"
|
432 |
+
numpy = ">=1.17"
|
433 |
+
packaging = ">=20.0"
|
434 |
+
pyyaml = ">=5.1"
|
435 |
+
regex = "!=2019.12.17"
|
436 |
+
requests = "*"
|
437 |
+
tokenizers = ">=0.11.1,<0.11.3 || >0.11.3,<0.13"
|
438 |
+
tqdm = ">=4.27"
|
439 |
+
|
440 |
+
[package.extras]
|
441 |
+
accelerate = ["accelerate (>=0.10.0)"]
|
442 |
+
all = ["tensorflow (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text", "torch (>=1.0)", "jax (>=0.2.8,!=0.3.2,<=0.3.6)", "jaxlib (>=0.1.65,<=0.3.6)", "flax (>=0.4.1)", "optax (>=0.0.8)", "sentencepiece (>=0.1.91,!=0.1.92)", "protobuf (<=3.20.1)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer", "pillow", "optuna", "ray", "sigopt", "timm", "codecarbon (==1.2.0)", "accelerate (>=0.10.0)"]
|
443 |
+
audio = ["librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
444 |
+
codecarbon = ["codecarbon (==1.2.0)"]
|
445 |
+
deepspeed = ["deepspeed (>=0.6.5)", "accelerate (>=0.10.0)"]
|
446 |
+
deepspeed-testing = ["deepspeed (>=0.6.5)", "accelerate (>=0.10.0)", "pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil", "datasets", "dill (<0.3.5)", "evaluate (>=0.2.0)", "pytest-timeout", "black (==22.3)", "sacrebleu (>=1.4.12,<2.0.0)", "rouge-score (!=0.0.7,!=0.0.8,!=0.1,!=0.1.1)", "nltk", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)", "protobuf (<=3.20.1)", "sacremoses", "rjieba", "faiss-cpu", "cookiecutter (==1.7.3)", "optuna"]
|
447 |
+
dev = ["tensorflow (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text", "torch (>=1.0)", "jax (>=0.2.8,!=0.3.2,<=0.3.6)", "jaxlib (>=0.1.65,<=0.3.6)", "flax (>=0.4.1)", "optax (>=0.0.8)", "sentencepiece (>=0.1.91,!=0.1.92)", "protobuf (<=3.20.1)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer", "pillow", "optuna", "ray", "sigopt", "timm", "codecarbon (==1.2.0)", "accelerate (>=0.10.0)", "pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil", "datasets", "dill (<0.3.5)", "evaluate (>=0.2.0)", "pytest-timeout", "black (==22.3)", "sacrebleu (>=1.4.12,<2.0.0)", "rouge-score (!=0.0.7,!=0.0.8,!=0.1,!=0.1.1)", "nltk", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)", "sacremoses", "rjieba", "faiss-cpu", "cookiecutter (==1.7.3)", "isort (>=5.5.4)", "flake8 (>=3.8.3)", "fugashi (>=1.0)", "ipadic (>=1.0.0,<2.0)", "unidic-lite (>=1.0.7)", "unidic (>=1.0.2)", "hf-doc-builder", "scikit-learn"]
|
448 |
+
dev-tensorflow = ["pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil", "datasets", "dill (<0.3.5)", "evaluate (>=0.2.0)", "pytest-timeout", "black (==22.3)", "sacrebleu (>=1.4.12,<2.0.0)", "rouge-score (!=0.0.7,!=0.0.8,!=0.1,!=0.1.1)", "nltk", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)", "protobuf (<=3.20.1)", "sacremoses", "rjieba", "faiss-cpu", "cookiecutter (==1.7.3)", "tensorflow (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text", "sentencepiece (>=0.1.91,!=0.1.92)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "pillow", "isort (>=5.5.4)", "flake8 (>=3.8.3)", "hf-doc-builder", "scikit-learn", "onnxruntime (>=1.4.0)", "onnxruntime-tools (>=1.4.2)", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
449 |
+
dev-torch = ["pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil", "datasets", "dill (<0.3.5)", "evaluate (>=0.2.0)", "pytest-timeout", "black (==22.3)", "sacrebleu (>=1.4.12,<2.0.0)", "rouge-score (!=0.0.7,!=0.0.8,!=0.1,!=0.1.1)", "nltk", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)", "protobuf (<=3.20.1)", "sacremoses", "rjieba", "faiss-cpu", "cookiecutter (==1.7.3)", "torch (>=1.0)", "sentencepiece (>=0.1.91,!=0.1.92)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer", "pillow", "optuna", "ray", "sigopt", "timm", "codecarbon (==1.2.0)", "isort (>=5.5.4)", "flake8 (>=3.8.3)", "fugashi (>=1.0)", "ipadic (>=1.0.0,<2.0)", "unidic-lite (>=1.0.7)", "unidic (>=1.0.2)", "hf-doc-builder", "scikit-learn", "onnxruntime (>=1.4.0)", "onnxruntime-tools (>=1.4.2)"]
|
450 |
+
docs = ["tensorflow (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text", "torch (>=1.0)", "jax (>=0.2.8,!=0.3.2,<=0.3.6)", "jaxlib (>=0.1.65,<=0.3.6)", "flax (>=0.4.1)", "optax (>=0.0.8)", "sentencepiece (>=0.1.91,!=0.1.92)", "protobuf (<=3.20.1)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer", "pillow", "optuna", "ray", "sigopt", "timm", "codecarbon (==1.2.0)", "accelerate (>=0.10.0)", "hf-doc-builder"]
|
451 |
+
docs_specific = ["hf-doc-builder"]
|
452 |
+
fairscale = ["fairscale (>0.3)"]
|
453 |
+
flax = ["jax (>=0.2.8,!=0.3.2,<=0.3.6)", "jaxlib (>=0.1.65,<=0.3.6)", "flax (>=0.4.1)", "optax (>=0.0.8)"]
|
454 |
+
flax-speech = ["librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
455 |
+
ftfy = ["ftfy"]
|
456 |
+
integrations = ["optuna", "ray", "sigopt"]
|
457 |
+
ja = ["fugashi (>=1.0)", "ipadic (>=1.0.0,<2.0)", "unidic-lite (>=1.0.7)", "unidic (>=1.0.2)"]
|
458 |
+
modelcreation = ["cookiecutter (==1.7.3)"]
|
459 |
+
onnx = ["onnxconverter-common", "tf2onnx", "onnxruntime (>=1.4.0)", "onnxruntime-tools (>=1.4.2)"]
|
460 |
+
onnxruntime = ["onnxruntime (>=1.4.0)", "onnxruntime-tools (>=1.4.2)"]
|
461 |
+
optuna = ["optuna"]
|
462 |
+
quality = ["black (==22.3)", "isort (>=5.5.4)", "flake8 (>=3.8.3)", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)"]
|
463 |
+
ray = ["ray"]
|
464 |
+
retrieval = ["faiss-cpu", "datasets"]
|
465 |
+
sagemaker = ["sagemaker (>=2.31.0)"]
|
466 |
+
sentencepiece = ["sentencepiece (>=0.1.91,!=0.1.92)", "protobuf (<=3.20.1)"]
|
467 |
+
serving = ["pydantic", "uvicorn", "fastapi", "starlette"]
|
468 |
+
sigopt = ["sigopt"]
|
469 |
+
sklearn = ["scikit-learn"]
|
470 |
+
speech = ["torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
471 |
+
testing = ["pytest", "pytest-xdist", "timeout-decorator", "parameterized", "psutil", "datasets", "dill (<0.3.5)", "evaluate (>=0.2.0)", "pytest-timeout", "black (==22.3)", "sacrebleu (>=1.4.12,<2.0.0)", "rouge-score (!=0.0.7,!=0.0.8,!=0.1,!=0.1.1)", "nltk", "GitPython (<3.1.19)", "hf-doc-builder (>=0.3.0)", "protobuf (<=3.20.1)", "sacremoses", "rjieba", "faiss-cpu", "cookiecutter (==1.7.3)"]
|
472 |
+
tf = ["tensorflow (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text"]
|
473 |
+
tf-cpu = ["tensorflow-cpu (>=2.3)", "onnxconverter-common", "tf2onnx", "tensorflow-text"]
|
474 |
+
tf-speech = ["librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
475 |
+
timm = ["timm"]
|
476 |
+
tokenizers = ["tokenizers (>=0.11.1,!=0.11.3,<0.13)"]
|
477 |
+
torch = ["torch (>=1.0)"]
|
478 |
+
torch-speech = ["torchaudio", "librosa", "pyctcdecode (>=0.3.0)", "phonemizer"]
|
479 |
+
torchhub = ["filelock", "huggingface-hub (>=0.9.0,<1.0)", "importlib-metadata", "numpy (>=1.17)", "packaging (>=20.0)", "protobuf (<=3.20.1)", "regex (!=2019.12.17)", "requests", "sentencepiece (>=0.1.91,!=0.1.92)", "torch (>=1.0)", "tokenizers (>=0.11.1,!=0.11.3,<0.13)", "tqdm (>=4.27)"]
|
480 |
+
vision = ["pillow"]
|
481 |
+
|
482 |
+
[[package]]
|
483 |
+
name = "typing-extensions"
|
484 |
+
version = "4.4.0"
|
485 |
+
description = "Backported and Experimental Type Hints for Python 3.7+"
|
486 |
+
category = "main"
|
487 |
+
optional = false
|
488 |
+
python-versions = ">=3.7"
|
489 |
+
|
490 |
+
[[package]]
|
491 |
+
name = "urllib3"
|
492 |
+
version = "1.26.12"
|
493 |
+
description = "HTTP library with thread-safe connection pooling, file post, and more."
|
494 |
+
category = "main"
|
495 |
+
optional = false
|
496 |
+
python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*, <4"
|
497 |
+
|
498 |
+
[package.extras]
|
499 |
+
brotli = ["brotlicffi (>=0.8.0)", "brotli (>=1.0.9)", "brotlipy (>=0.6.0)"]
|
500 |
+
secure = ["pyOpenSSL (>=0.14)", "cryptography (>=1.3.4)", "idna (>=2.0.0)", "certifi", "urllib3-secure-extra", "ipaddress"]
|
501 |
+
socks = ["PySocks (>=1.5.6,!=1.5.7,<2.0)"]
|
502 |
+
|
503 |
+
[[package]]
|
504 |
+
name = "wcwidth"
|
505 |
+
version = "0.2.5"
|
506 |
+
description = "Measures the displayed width of unicode strings in a terminal"
|
507 |
+
category = "dev"
|
508 |
+
optional = false
|
509 |
+
python-versions = "*"
|
510 |
+
|
511 |
+
[[package]]
|
512 |
+
name = "werkzeug"
|
513 |
+
version = "2.2.2"
|
514 |
+
description = "The comprehensive WSGI web application library."
|
515 |
+
category = "main"
|
516 |
+
optional = false
|
517 |
+
python-versions = ">=3.7"
|
518 |
+
|
519 |
+
[package.dependencies]
|
520 |
+
MarkupSafe = ">=2.1.1"
|
521 |
+
|
522 |
+
[package.extras]
|
523 |
+
watchdog = ["watchdog"]
|
524 |
+
|
525 |
+
[[package]]
|
526 |
+
name = "zipp"
|
527 |
+
version = "3.8.1"
|
528 |
+
description = "Backport of pathlib-compatible object wrapper for zip files"
|
529 |
+
category = "main"
|
530 |
+
optional = false
|
531 |
+
python-versions = ">=3.7"
|
532 |
+
|
533 |
+
[package.extras]
|
534 |
+
docs = ["sphinx", "jaraco.packaging (>=9)", "rst.linker (>=1.9)", "jaraco.tidelift (>=1.4)"]
|
535 |
+
testing = ["pytest (>=6)", "pytest-checkdocs (>=2.4)", "pytest-flake8", "pytest-cov", "pytest-enabler (>=1.3)", "jaraco.itertools", "func-timeout", "pytest-black (>=0.3.7)", "pytest-mypy (>=0.9.1)"]
|
536 |
+
|
537 |
+
[metadata]
|
538 |
+
lock-version = "1.1"
|
539 |
+
python-versions = "^3.9"
|
540 |
+
content-hash = "60e4d68e194fc67ea38fbdd192e107a9eea2d7a809a662fc121e8011e613f0fc"
|
541 |
+
|
542 |
+
[metadata.files]
|
543 |
+
atomicwrites = []
|
544 |
+
attrs = []
|
545 |
+
black = []
|
546 |
+
certifi = []
|
547 |
+
charset-normalizer = []
|
548 |
+
click = [
|
549 |
+
{file = "click-8.1.3-py3-none-any.whl", hash = "sha256:bb4d8133cb15a609f44e8213d9b391b0809795062913b383c62be0ee95b1db48"},
|
550 |
+
{file = "click-8.1.3.tar.gz", hash = "sha256:7682dc8afb30297001674575ea00d1814d808d6a36af415a82bd481d37ba7b8e"},
|
551 |
+
]
|
552 |
+
colorama = []
|
553 |
+
diffusers = []
|
554 |
+
filelock = []
|
555 |
+
flask = []
|
556 |
+
huggingface-hub = []
|
557 |
+
idna = []
|
558 |
+
importlib-metadata = []
|
559 |
+
itsdangerous = []
|
560 |
+
jinja2 = [
|
561 |
+
{file = "Jinja2-3.1.2-py3-none-any.whl", hash = "sha256:6088930bfe239f0e6710546ab9c19c9ef35e29792895fed6e6e31a023a182a61"},
|
562 |
+
{file = "Jinja2-3.1.2.tar.gz", hash = "sha256:31351a702a408a9e7595a8fc6150fc3f43bb6bf7e319770cbc0db9df9437e852"},
|
563 |
+
]
|
564 |
+
markupsafe = [
|
565 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:86b1f75c4e7c2ac2ccdaec2b9022845dbb81880ca318bb7a0a01fbf7813e3812"},
|
566 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:f121a1420d4e173a5d96e47e9a0c0dcff965afdf1626d28de1460815f7c4ee7a"},
|
567 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:a49907dd8420c5685cfa064a1335b6754b74541bbb3706c259c02ed65b644b3e"},
|
568 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:10c1bfff05d95783da83491be968e8fe789263689c02724e0c691933c52994f5"},
|
569 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:b7bd98b796e2b6553da7225aeb61f447f80a1ca64f41d83612e6139ca5213aa4"},
|
570 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-musllinux_1_1_aarch64.whl", hash = "sha256:b09bf97215625a311f669476f44b8b318b075847b49316d3e28c08e41a7a573f"},
|
571 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:694deca8d702d5db21ec83983ce0bb4b26a578e71fbdbd4fdcd387daa90e4d5e"},
|
572 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:efc1913fd2ca4f334418481c7e595c00aad186563bbc1ec76067848c7ca0a933"},
|
573 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-win32.whl", hash = "sha256:4a33dea2b688b3190ee12bd7cfa29d39c9ed176bda40bfa11099a3ce5d3a7ac6"},
|
574 |
+
{file = "MarkupSafe-2.1.1-cp310-cp310-win_amd64.whl", hash = "sha256:dda30ba7e87fbbb7eab1ec9f58678558fd9a6b8b853530e176eabd064da81417"},
|
575 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:671cd1187ed5e62818414afe79ed29da836dde67166a9fac6d435873c44fdd02"},
|
576 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:3799351e2336dc91ea70b034983ee71cf2f9533cdff7c14c90ea126bfd95d65a"},
|
577 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:e72591e9ecd94d7feb70c1cbd7be7b3ebea3f548870aa91e2732960fa4d57a37"},
|
578 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:6fbf47b5d3728c6aea2abb0589b5d30459e369baa772e0f37a0320185e87c980"},
|
579 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-musllinux_1_1_aarch64.whl", hash = "sha256:d5ee4f386140395a2c818d149221149c54849dfcfcb9f1debfe07a8b8bd63f9a"},
|
580 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-musllinux_1_1_i686.whl", hash = "sha256:bcb3ed405ed3222f9904899563d6fc492ff75cce56cba05e32eff40e6acbeaa3"},
|
581 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-musllinux_1_1_x86_64.whl", hash = "sha256:e1c0b87e09fa55a220f058d1d49d3fb8df88fbfab58558f1198e08c1e1de842a"},
|
582 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-win32.whl", hash = "sha256:8dc1c72a69aa7e082593c4a203dcf94ddb74bb5c8a731e4e1eb68d031e8498ff"},
|
583 |
+
{file = "MarkupSafe-2.1.1-cp37-cp37m-win_amd64.whl", hash = "sha256:97a68e6ada378df82bc9f16b800ab77cbf4b2fada0081794318520138c088e4a"},
|
584 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-macosx_10_9_universal2.whl", hash = "sha256:e8c843bbcda3a2f1e3c2ab25913c80a3c5376cd00c6e8c4a86a89a28c8dc5452"},
|
585 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:0212a68688482dc52b2d45013df70d169f542b7394fc744c02a57374a4207003"},
|
586 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:8e576a51ad59e4bfaac456023a78f6b5e6e7651dcd383bcc3e18d06f9b55d6d1"},
|
587 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:4b9fe39a2ccc108a4accc2676e77da025ce383c108593d65cc909add5c3bd601"},
|
588 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:96e37a3dc86e80bf81758c152fe66dbf60ed5eca3d26305edf01892257049925"},
|
589 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-musllinux_1_1_aarch64.whl", hash = "sha256:6d0072fea50feec76a4c418096652f2c3238eaa014b2f94aeb1d56a66b41403f"},
|
590 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-musllinux_1_1_i686.whl", hash = "sha256:089cf3dbf0cd6c100f02945abeb18484bd1ee57a079aefd52cffd17fba910b88"},
|
591 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-musllinux_1_1_x86_64.whl", hash = "sha256:6a074d34ee7a5ce3effbc526b7083ec9731bb3cbf921bbe1d3005d4d2bdb3a63"},
|
592 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-win32.whl", hash = "sha256:421be9fbf0ffe9ffd7a378aafebbf6f4602d564d34be190fc19a193232fd12b1"},
|
593 |
+
{file = "MarkupSafe-2.1.1-cp38-cp38-win_amd64.whl", hash = "sha256:fc7b548b17d238737688817ab67deebb30e8073c95749d55538ed473130ec0c7"},
|
594 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-macosx_10_9_universal2.whl", hash = "sha256:e04e26803c9c3851c931eac40c695602c6295b8d432cbe78609649ad9bd2da8a"},
|
595 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:b87db4360013327109564f0e591bd2a3b318547bcef31b468a92ee504d07ae4f"},
|
596 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:99a2a507ed3ac881b975a2976d59f38c19386d128e7a9a18b7df6fff1fd4c1d6"},
|
597 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:56442863ed2b06d19c37f94d999035e15ee982988920e12a5b4ba29b62ad1f77"},
|
598 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:3ce11ee3f23f79dbd06fb3d63e2f6af7b12db1d46932fe7bd8afa259a5996603"},
|
599 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-musllinux_1_1_aarch64.whl", hash = "sha256:33b74d289bd2f5e527beadcaa3f401e0df0a89927c1559c8566c066fa4248ab7"},
|
600 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:43093fb83d8343aac0b1baa75516da6092f58f41200907ef92448ecab8825135"},
|
601 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:8e3dcf21f367459434c18e71b2a9532d96547aef8a871872a5bd69a715c15f96"},
|
602 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-win32.whl", hash = "sha256:d4306c36ca495956b6d568d276ac11fdd9c30a36f1b6eb928070dc5360b22e1c"},
|
603 |
+
{file = "MarkupSafe-2.1.1-cp39-cp39-win_amd64.whl", hash = "sha256:46d00d6cfecdde84d40e572d63735ef81423ad31184100411e6e3388d405e247"},
|
604 |
+
{file = "MarkupSafe-2.1.1.tar.gz", hash = "sha256:7f91197cc9e48f989d12e4e6fbc46495c446636dfc81b9ccf50bb0ec74b91d4b"},
|
605 |
+
]
|
606 |
+
more-itertools = []
|
607 |
+
mypy-extensions = [
|
608 |
+
{file = "mypy_extensions-0.4.3-py2.py3-none-any.whl", hash = "sha256:090fedd75945a69ae91ce1303b5824f428daf5a028d2f6ab8a299250a846f15d"},
|
609 |
+
{file = "mypy_extensions-0.4.3.tar.gz", hash = "sha256:2d82818f5bb3e369420cb3c4060a7970edba416647068eb4c5343488a6c604a8"},
|
610 |
+
]
|
611 |
+
numpy = []
|
612 |
+
packaging = [
|
613 |
+
{file = "packaging-21.3-py3-none-any.whl", hash = "sha256:ef103e05f519cdc783ae24ea4e2e0f508a9c99b2d4969652eed6a2e1ea5bd522"},
|
614 |
+
{file = "packaging-21.3.tar.gz", hash = "sha256:dd47c42927d89ab911e606518907cc2d3a1f38bbd026385970643f9c5b8ecfeb"},
|
615 |
+
]
|
616 |
+
pathspec = []
|
617 |
+
pillow = []
|
618 |
+
platformdirs = [
|
619 |
+
{file = "platformdirs-2.5.2-py3-none-any.whl", hash = "sha256:027d8e83a2d7de06bbac4e5ef7e023c02b863d7ea5d079477e722bb41ab25788"},
|
620 |
+
{file = "platformdirs-2.5.2.tar.gz", hash = "sha256:58c8abb07dcb441e6ee4b11d8df0ac856038f944ab98b7be6b27b2a3c7feef19"},
|
621 |
+
]
|
622 |
+
pluggy = [
|
623 |
+
{file = "pluggy-0.13.1-py2.py3-none-any.whl", hash = "sha256:966c145cd83c96502c3c3868f50408687b38434af77734af1e9ca461a4081d2d"},
|
624 |
+
{file = "pluggy-0.13.1.tar.gz", hash = "sha256:15b2acde666561e1298d71b523007ed7364de07029219b604cf808bfa1c765b0"},
|
625 |
+
]
|
626 |
+
py = [
|
627 |
+
{file = "py-1.11.0-py2.py3-none-any.whl", hash = "sha256:607c53218732647dff4acdfcd50cb62615cedf612e72d1724fb1a0cc6405b378"},
|
628 |
+
{file = "py-1.11.0.tar.gz", hash = "sha256:51c75c4126074b472f746a24399ad32f6053d1b34b68d2fa41e558e6f4a98719"},
|
629 |
+
]
|
630 |
+
pyparsing = [
|
631 |
+
{file = "pyparsing-3.0.9-py3-none-any.whl", hash = "sha256:5026bae9a10eeaefb61dab2f09052b9f4307d44aee4eda64b309723d8d206bbc"},
|
632 |
+
{file = "pyparsing-3.0.9.tar.gz", hash = "sha256:2b020ecf7d21b687f219b71ecad3631f644a47f01403fa1d1036b0c6416d70fb"},
|
633 |
+
]
|
634 |
+
pytest = [
|
635 |
+
{file = "pytest-5.4.3-py3-none-any.whl", hash = "sha256:5c0db86b698e8f170ba4582a492248919255fcd4c79b1ee64ace34301fb589a1"},
|
636 |
+
{file = "pytest-5.4.3.tar.gz", hash = "sha256:7979331bfcba207414f5e1263b5a0f8f521d0f457318836a7355531ed1a4c7d8"},
|
637 |
+
]
|
638 |
+
pyyaml = [
|
639 |
+
{file = "PyYAML-6.0-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:d4db7c7aef085872ef65a8fd7d6d09a14ae91f691dec3e87ee5ee0539d516f53"},
|
640 |
+
{file = "PyYAML-6.0-cp310-cp310-macosx_11_0_arm64.whl", hash = "sha256:9df7ed3b3d2e0ecfe09e14741b857df43adb5a3ddadc919a2d94fbdf78fea53c"},
|
641 |
+
{file = "PyYAML-6.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:77f396e6ef4c73fdc33a9157446466f1cff553d979bd00ecb64385760c6babdc"},
|
642 |
+
{file = "PyYAML-6.0-cp310-cp310-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:a80a78046a72361de73f8f395f1f1e49f956c6be882eed58505a15f3e430962b"},
|
643 |
+
{file = "PyYAML-6.0-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:f84fbc98b019fef2ee9a1cb3ce93e3187a6df0b2538a651bfb890254ba9f90b5"},
|
644 |
+
{file = "PyYAML-6.0-cp310-cp310-win32.whl", hash = "sha256:2cd5df3de48857ed0544b34e2d40e9fac445930039f3cfe4bcc592a1f836d513"},
|
645 |
+
{file = "PyYAML-6.0-cp310-cp310-win_amd64.whl", hash = "sha256:daf496c58a8c52083df09b80c860005194014c3698698d1a57cbcfa182142a3a"},
|
646 |
+
{file = "PyYAML-6.0-cp36-cp36m-macosx_10_9_x86_64.whl", hash = "sha256:897b80890765f037df3403d22bab41627ca8811ae55e9a722fd0392850ec4d86"},
|
647 |
+
{file = "PyYAML-6.0-cp36-cp36m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:50602afada6d6cbfad699b0c7bb50d5ccffa7e46a3d738092afddc1f9758427f"},
|
648 |
+
{file = "PyYAML-6.0-cp36-cp36m-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:48c346915c114f5fdb3ead70312bd042a953a8ce5c7106d5bfb1a5254e47da92"},
|
649 |
+
{file = "PyYAML-6.0-cp36-cp36m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:98c4d36e99714e55cfbaaee6dd5badbc9a1ec339ebfc3b1f52e293aee6bb71a4"},
|
650 |
+
{file = "PyYAML-6.0-cp36-cp36m-win32.whl", hash = "sha256:0283c35a6a9fbf047493e3a0ce8d79ef5030852c51e9d911a27badfde0605293"},
|
651 |
+
{file = "PyYAML-6.0-cp36-cp36m-win_amd64.whl", hash = "sha256:07751360502caac1c067a8132d150cf3d61339af5691fe9e87803040dbc5db57"},
|
652 |
+
{file = "PyYAML-6.0-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:819b3830a1543db06c4d4b865e70ded25be52a2e0631ccd2f6a47a2822f2fd7c"},
|
653 |
+
{file = "PyYAML-6.0-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:473f9edb243cb1935ab5a084eb238d842fb8f404ed2193a915d1784b5a6b5fc0"},
|
654 |
+
{file = "PyYAML-6.0-cp37-cp37m-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:0ce82d761c532fe4ec3f87fc45688bdd3a4c1dc5e0b4a19814b9009a29baefd4"},
|
655 |
+
{file = "PyYAML-6.0-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:231710d57adfd809ef5d34183b8ed1eeae3f76459c18fb4a0b373ad56bedcdd9"},
|
656 |
+
{file = "PyYAML-6.0-cp37-cp37m-win32.whl", hash = "sha256:c5687b8d43cf58545ade1fe3e055f70eac7a5a1a0bf42824308d868289a95737"},
|
657 |
+
{file = "PyYAML-6.0-cp37-cp37m-win_amd64.whl", hash = "sha256:d15a181d1ecd0d4270dc32edb46f7cb7733c7c508857278d3d378d14d606db2d"},
|
658 |
+
{file = "PyYAML-6.0-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:0b4624f379dab24d3725ffde76559cff63d9ec94e1736b556dacdfebe5ab6d4b"},
|
659 |
+
{file = "PyYAML-6.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:213c60cd50106436cc818accf5baa1aba61c0189ff610f64f4a3e8c6726218ba"},
|
660 |
+
{file = "PyYAML-6.0-cp38-cp38-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:9fa600030013c4de8165339db93d182b9431076eb98eb40ee068700c9c813e34"},
|
661 |
+
{file = "PyYAML-6.0-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:277a0ef2981ca40581a47093e9e2d13b3f1fbbeffae064c1d21bfceba2030287"},
|
662 |
+
{file = "PyYAML-6.0-cp38-cp38-win32.whl", hash = "sha256:d4eccecf9adf6fbcc6861a38015c2a64f38b9d94838ac1810a9023a0609e1b78"},
|
663 |
+
{file = "PyYAML-6.0-cp38-cp38-win_amd64.whl", hash = "sha256:1e4747bc279b4f613a09eb64bba2ba602d8a6664c6ce6396a4d0cd413a50ce07"},
|
664 |
+
{file = "PyYAML-6.0-cp39-cp39-macosx_10_9_x86_64.whl", hash = "sha256:055d937d65826939cb044fc8c9b08889e8c743fdc6a32b33e2390f66013e449b"},
|
665 |
+
{file = "PyYAML-6.0-cp39-cp39-macosx_11_0_arm64.whl", hash = "sha256:e61ceaab6f49fb8bdfaa0f92c4b57bcfbea54c09277b1b4f7ac376bfb7a7c174"},
|
666 |
+
{file = "PyYAML-6.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:d67d839ede4ed1b28a4e8909735fc992a923cdb84e618544973d7dfc71540803"},
|
667 |
+
{file = "PyYAML-6.0-cp39-cp39-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:cba8c411ef271aa037d7357a2bc8f9ee8b58b9965831d9e51baf703280dc73d3"},
|
668 |
+
{file = "PyYAML-6.0-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:40527857252b61eacd1d9af500c3337ba8deb8fc298940291486c465c8b46ec0"},
|
669 |
+
{file = "PyYAML-6.0-cp39-cp39-win32.whl", hash = "sha256:b5b9eccad747aabaaffbc6064800670f0c297e52c12754eb1d976c57e4f74dcb"},
|
670 |
+
{file = "PyYAML-6.0-cp39-cp39-win_amd64.whl", hash = "sha256:b3d267842bf12586ba6c734f89d1f5b871df0273157918b0ccefa29deb05c21c"},
|
671 |
+
{file = "PyYAML-6.0.tar.gz", hash = "sha256:68fb519c14306fec9720a2a5b45bc9f0c8d1b9c72adf45c37baedfcd949c35a2"},
|
672 |
+
]
|
673 |
+
regex = []
|
674 |
+
requests = []
|
675 |
+
tokenizers = []
|
676 |
+
tomli = [
|
677 |
+
{file = "tomli-2.0.1-py3-none-any.whl", hash = "sha256:939de3e7a6161af0c887ef91b7d41a53e7c5a1ca976325f429cb46ea9bc30ecc"},
|
678 |
+
{file = "tomli-2.0.1.tar.gz", hash = "sha256:de526c12914f0c550d15924c62d72abc48d6fe7364aa87328337a31007fe8a4f"},
|
679 |
+
]
|
680 |
+
torch = []
|
681 |
+
tqdm = []
|
682 |
+
transformers = []
|
683 |
+
typing-extensions = []
|
684 |
+
urllib3 = []
|
685 |
+
wcwidth = [
|
686 |
+
{file = "wcwidth-0.2.5-py2.py3-none-any.whl", hash = "sha256:beb4802a9cebb9144e99086eff703a642a13d6a0052920003a230f3294bbe784"},
|
687 |
+
{file = "wcwidth-0.2.5.tar.gz", hash = "sha256:c4d647b99872929fdb7bdcaa4fbe7f01413ed3d98077df798530e5b04f116c83"},
|
688 |
+
]
|
689 |
+
werkzeug = []
|
690 |
+
zipp = []
|
pyproject.toml
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[tool.poetry]
|
2 |
+
name = "text-brush"
|
3 |
+
version = "0.1.0"
|
4 |
+
description = ""
|
5 |
+
authors = ["Iliya Zhechev <iliya.zhechev@gmail.com>"]
|
6 |
+
|
7 |
+
[tool.poetry.dependencies]
|
8 |
+
python = "^3.9"
|
9 |
+
Flask = "^2.2.2"
|
10 |
+
torch = "^1.12.1"
|
11 |
+
diffusers = "^0.4.1"
|
12 |
+
transformers = "^4.22.2"
|
13 |
+
|
14 |
+
[tool.poetry.dev-dependencies]
|
15 |
+
pytest = "^5.2"
|
16 |
+
black = {version = "^22.10.0", allow-prereleases = true}
|
17 |
+
|
18 |
+
[build-system]
|
19 |
+
requires = ["poetry-core>=1.0.0"]
|
20 |
+
build-backend = "poetry.core.masonry.api"
|
tests/__init__.py
ADDED
File without changes
|
tests/test_text_brush.py
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from text_brush import __version__
|
2 |
+
|
3 |
+
|
4 |
+
def test_version():
|
5 |
+
assert __version__ == '0.1.0'
|
text_brush/.gitignore
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
*.pyc
|
text_brush/__init__.py
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
__version__ = '0.1.0'
|
text_brush/app.py
ADDED
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from flask import Flask, send_file, request
|
2 |
+
|
3 |
+
import torch
|
4 |
+
from torch import autocast
|
5 |
+
from diffusers import StableDiffusionPipeline
|
6 |
+
from io import BytesIO
|
7 |
+
|
8 |
+
app = Flask(__name__)
|
9 |
+
|
10 |
+
|
11 |
+
model_id = "CompVis/stable-diffusion-v1-4"
|
12 |
+
device = "cuda"
|
13 |
+
|
14 |
+
|
15 |
+
pipe = StableDiffusionPipeline.from_pretrained(
|
16 |
+
model_id,
|
17 |
+
torch_dtype=torch.float16,
|
18 |
+
revision="fp16",
|
19 |
+
use_auth_token=True,
|
20 |
+
)
|
21 |
+
|
22 |
+
pipe = pipe.to(device)
|
23 |
+
|
24 |
+
|
25 |
+
def serve_pil_image(pil_img):
|
26 |
+
img_io = BytesIO()
|
27 |
+
pil_img.save(img_io, "JPEG", quality=70)
|
28 |
+
img_io.seek(0)
|
29 |
+
return send_file(img_io, mimetype="image/jpeg")
|
30 |
+
|
31 |
+
|
32 |
+
@app.route("/")
|
33 |
+
def hello_world():
|
34 |
+
return "<p>Hello, World!</p>"
|
35 |
+
|
36 |
+
|
37 |
+
@app.route("/generate-image")
|
38 |
+
def generate_image():
|
39 |
+
prompt = request.args.get(
|
40 |
+
"prompt",
|
41 |
+
default="a photo of an astronaut riding a horse on mars",
|
42 |
+
type=str,
|
43 |
+
)
|
44 |
+
steps = request.args.get("steps", default=15, type=int)
|
45 |
+
seed = request.args.get("seed", default=1024, type=int)
|
46 |
+
|
47 |
+
generator = torch.Generator("cuda").manual_seed(seed)
|
48 |
+
with autocast(device):
|
49 |
+
image = pipe(
|
50 |
+
prompt,
|
51 |
+
guidance_scale=7.5,
|
52 |
+
num_inference_steps=steps,
|
53 |
+
generator=generator,
|
54 |
+
).images[0]
|
55 |
+
|
56 |
+
return serve_pil_image(image)
|
57 |
+
|
58 |
+
|
59 |
+
if __name__ == "__main__":
|
60 |
+
app.run(
|
61 |
+
host="0.0.0.0",
|
62 |
+
port=5000,
|
63 |
+
debug=True,
|
64 |
+
)
|