Merge branch 'main' of github.com:ichko/text-brush

server/poetry.lock

@@ -137,6 +137,18 @@ Werkzeug = ">=2.2.2"
 async = ["asgiref (>=3.2)"]
 dotenv = ["python-dotenv"]
 
+[[package]]
+name = "flask-cors"
+version = "3.0.10"
+description = "A Flask extension adding a decorator for CORS support"
+category = "main"
+optional = false
+python-versions = "*"
+
+[package.dependencies]
+Flask = ">=0.9"
+Six = "*"
+
 [[package]]
 name = "huggingface-hub"
 version = "0.10.0"
@@ -370,6 +382,14 @@ urllib3 = ">=1.21.1,<1.27"
 socks = ["PySocks (>=1.5.6,!=1.5.7)"]
 use_chardet_on_py3 = ["chardet (>=3.0.2,<6)"]
 
+[[package]]
+name = "six"
+version = "1.16.0"
+description = "Python 2 and 3 compatibility utilities"
+category = "main"
+optional = false
+python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*"
+
 [[package]]
 name = "tokenizers"
 version = "0.12.1"
@@ -537,7 +557,7 @@ testing = ["pytest (>=6)", "pytest-checkdocs (>=2.4)", "pytest-flake8", "pytest-
 [metadata]
 lock-version = "1.1"
 python-versions = "^3.9"
-content-hash = "60e4d68e194fc67ea38fbdd192e107a9eea2d7a809a662fc121e8011e613f0fc"
+content-hash = "36266ae2bd8cb7f60f9b8b2ac733a344d2e99e1e26890e9b4e6b4f844c2c7aa8"
 
 [metadata.files]
 atomicwrites = []
@@ -553,6 +573,7 @@ colorama = []
 diffusers = []
 filelock = []
 flask = []
+flask-cors = []
 huggingface-hub = []
 idna = []
 importlib-metadata = []
@@ -672,6 +693,10 @@ pyyaml = [
 ]
 regex = []
 requests = []
+six = [
+    {file = "six-1.16.0-py2.py3-none-any.whl", hash = "sha256:8abb2f1d86890a2dfb989f9a77cfcfd3e47c2a354b01111771326f8aa26e0254"},
+    {file = "six-1.16.0.tar.gz", hash = "sha256:1e61c37477a1626458e36f7b1d82aa5c9b094fa4802892072e49de9c60c4c926"},
+]
 tokenizers = []
 tomli = [
     {file = "tomli-2.0.1-py3-none-any.whl", hash = "sha256:939de3e7a6161af0c887ef91b7d41a53e7c5a1ca976325f429cb46ea9bc30ecc"},

server/pyproject.toml

@@ -10,6 +10,7 @@ Flask = "^2.2.2"
 torch = "^1.12.1"
 diffusers = "^0.4.1"
 transformers = "^4.22.2"
+Flask-Cors = "^3.0.10"
 
 [tool.poetry.dev-dependencies]
 pytest = "^5.2"

server/text_brush/app.py

@@ -4,8 +4,11 @@ import torch
 from torch import autocast
 from diffusers import StableDiffusionPipeline
 from io import BytesIO
+from flask_cors import CORS
+
 
 app = Flask(__name__)
+CORS(app)
 
 
 model_id = "CompVis/stable-diffusion-v1-4"