diff --git a/handler.py b/handler.py index 48021c7..b30a712 100644 --- a/handler.py +++ b/handler.py @@ -21,14 +21,9 @@ t = time.time() return inner -def load_demo_image(img, image_size, device): - if "http" in img_url: - # img_url = 'https://storage.googleapis.com/sfr-vision-language-research/BLIP/demo.jpg' - raw_image = Image.open(requests.get(img_url, stream=True).raw).convert('RGB') - else: - raw_image = Image.open(img_url).convert('RGB') - - w,h = raw_image.size +def load_demo_image(base64_str, image_size, device): + img = Image.open(base64_str) + raw_image = np.array(img.convert("RGB")) transform = transforms.Compose([ transforms.Resize((image_size,image_size),interpolation=InterpolationMode.BICUBIC), @@ -47,9 +42,9 @@ def handle(conf): - image_url = conf['Photo'] # value_type: str # description: some description + base64_str = conf['Photo'] # value_type: str # description: some description question = conf['Question'] - image = load_demo_image(image_url, image_size, device) + image = load_demo_image(base64_str, image_size, device) res = inference(model, image, question) print('Answer :', res) # add your code diff --git a/project_requirements.txt b/project_requirements.txt index c75110c..8186101 100644 --- a/project_requirements.txt +++ b/project_requirements.txt @@ -10,7 +10,7 @@ thinc==7.4.1 charset-normalizer==2.0.12 nltk==3.5 -transformers==4.15.0 +transformers==4.1.1 astunparse==1.6.3 PyWavelets==1.2.0 semantic-version==2.8.5 @@ -40,17 +40,16 @@ s3transfer==0.3.3 networkx==2.6.3 catalogue==1.0.0 -fairscale==0.4.4 +openpyxl==2.6.4 dm-tree==0.1.6 imageio==2.8.0 metakernel==0.28.2 opt-einsum==3.3.0 cachetools==3.1.1 smart-open==5.2.1 -timm==0.4.12 +timm==0.5.4 multipledispatch==0.6.0 argon2-cffi==21.3.0 -pycocoevalcap==1.2 tinycss2==1.1.1 graphviz==0.14 minio==5.0.10 @@ -81,7 +80,6 @@ cryptography==36.0.1 python-json-logger==2.0.2 ruamel.yaml==0.17.21 -pycocotools==2.0.4 pyglet==1.5.0 cymem==2.0.6 Shapely==1.7.0 @@ -101,15 +99,14 @@ xlrd==1.2.0 numpyencoder==0.3.0 copulas==0.3.3 -torch==1.11.0 +torch==1.4.0+cpu typeguard==2.13.3 pyOpenSSL==22.0.0 matplotlib-inline==0.1.3 tqdm==4.46.1 -torchvision==0.12.0 +torchvision==0.5.0+cpu wrapt==1.13.3 google-auth-oauthlib==0.4.6 -huggingface-hub==0.7.0 pycparser==2.21 filelock==3.6.0 botocore==1.19.25 @@ -141,5 +138,4 @@ pyrsistent==0.18.1 imgaug==0.4.0 asttokens==2.0.5 -tokenizers==0.10.3 -openpyxl==2.6.4 +tokenizers==0.9.4 diff --git a/reset_requirements.txt b/reset_requirements.txt deleted file mode 100644 index 63df11f..0000000 --- a/reset_requirements.txt +++ /dev/null @@ -1,224 +0,0 @@ -absl-py==0.9.0 -alembic==1.7.6 -argon2-cffi==21.3.0 -argon2-cffi-bindings==21.2.0 -astor==0.8.1 -asttokens==2.0.5 -astunparse==1.6.3 -async-generator==1.10 -attrs==19.3.0 -Augmentor==0.2.8 -backcall==0.2.0 -baytune==0.3.12 -bleach==4.1.0 -blis==0.4.1 -boto3==1.16.25 -botocore==1.19.25 -cachetools==3.1.1 -cairocffi==1.3.0 -CairoSVG==2.5.2 -calysto==1.0.6 -catalogue==1.0.0 -certifi==2021.10.8 -certipy==0.1.3 -cffi==1.15.0 -charset-normalizer==2.0.12 -click==8.0.4 -cloudpickle==1.2.2 -cmake==3.21.1 -configparser==5.2.0 -copulas==0.3.3 -cryptography==36.0.1 -cssselect2==0.4.1 -cycler==0.11.0 -cymem==2.0.6 -Cython==0.29.20 -debugpy==1.5.1 -decorator==5.1.1 -defusedxml==0.7.1 -distlib==0.3.4 -dlib==19.22.0 -dm-tree==0.1.6 -easydict==1.9 -en-core-web-sm @ https://files.momodel.cn/en_core_web_sm-2.3.0.tar.gz -entrypoints==0.4 -et-xmlfile==1.1.0 -fairscale==0.4.4 -filelock==3.6.0 -func-timeout==4.3.5 -future==0.18.2 -gast==0.3.3 -gensim==3.8.3 -google-auth==2.6.0 -google-auth-oauthlib==0.4.6 -google-pasta==0.2.0 -graphviz==0.14 -greenlet==1.1.2 -grpcio==1.29.0 -gym==0.15.7 -h5py==2.10.0 -huggingface-hub==0.7.0 -idna==3.3 -imageio==2.8.0 -imbalanced-learn==0.6.2 -imgaug==0.4.0 -importlib-metadata==4.11.1 -importlib-resources==5.4.0 -install==1.3.5 -ipdb==0.13.2 -ipykernel==6.9.1 -ipython==7.31.1 -ipython-genutils==0.2.0 -ipywidgets==7.4.0 -jdcal==1.4.1 -jedi==0.18.1 -jieba==0.42.1 -Jinja2==3.0.3 -jmespath==0.10.0 -joblib==1.1.0 -jsonschema==4.4.0 -jupyter-client==7.1.2 -jupyter-core==4.9.2 -jupyter-telemetry==0.1.0 -jupyterhub==1.4.2 -jupyterlab==1.0.0a1 -jupyterlab-server==0.2.0 -kanren==0.2.3 -Keras==2.4.3 -Keras-Preprocessing==1.1.2 -kiwisolver==1.3.2 -Mako==1.1.6 -Markdown==3.3.6 -MarkupSafe==2.1.0 -matplotlib==3.0.3 -matplotlib-inline==0.1.3 -metakernel==0.28.2 -mindspore @ https://ms-release.obs.cn-north-4.myhuaweicloud.com/1.0.0/MindSpore/cpu/ubuntu_x86/mindspore-1.0.0-cp37-cp37m-linux_x86_64.whl -minepy==1.2.4 -minio==5.0.10 -mistune==0.8.4 -mpmath==1.2.1 -multipledispatch==0.6.0 -murmurhash==1.0.6 -nbconvert==5.6.1 -nbformat==5.1.3 -nest-asyncio==1.5.4 -networkx==2.6.3 -nltk==3.5 -notebook==6.2.0 -numpy==1.18.5 -numpyencoder==0.3.0 -oauthlib==3.2.0 -opencv-python==4.5.1.48 -openpyxl==2.6.4 -opt-einsum==3.3.0 -packaging==21.3 -paddlepaddle==2.0.1 -pamela==1.0.0 -pandas==0.24.2 -pandocfilters==1.5.0 -parso==0.8.3 -pbr==5.8.1 -pexpect==4.8.0 -pickleshare==0.7.5 -Pillow==8.1.0 -plac==1.1.3 -platformdirs==2.5.1 -plotly==4.8.1 -portpicker==1.3.9 -preshed==3.0.6 -prometheus-client==0.13.1 -prompt-toolkit==3.0.28 -protobuf==3.19.4 -ptyprocess==0.7.0 -pyasn1==0.4.8 -pyasn1-modules==0.2.8 -PyAudio==0.2.11 -pycocoevalcap==1.2 -pycocotools==2.0.4 -pycparser==2.21 -pydot==1.4.1 -pygame==2.0.1 -pyglet==1.5.0 -Pygments==2.11.2 -pyOpenSSL==22.0.0 -pyparsing==3.0.7 -pyrsistent==0.18.1 -python-dateutil==2.8.2 -python-json-logger==2.0.2 -pytorch-pretrained-bert==0.6.2 -pytorch-transformers==1.2.0 -pytz==2021.3 -PyWavelets==1.2.0 -PyYAML==6.0 -pyzmq==22.3.0 -regex==2022.1.18 -requests==2.27.1 -requests-oauthlib==1.3.1 -retrying==1.3.3 -rouge==1.0.0 -rsa==4.8 -ruamel.yaml==0.17.21 -ruamel.yaml.clib==0.2.6 -s3transfer==0.3.3 -sacremoses==0.0.47 -scikit-image==0.15.0 -scikit-learn==0.22.2.post1 -scipy==1.3.3 -seaborn==0.10.1 -semantic-version==2.8.5 -Send2Trash==1.8.0 -sentencepiece==0.1.91 -Shapely==1.7.0 -six==1.16.0 -smart-open==5.2.1 -spacy==2.3.2 -SQLAlchemy==1.4.31 -srsly==1.0.5 -stevedore==3.5.0 -svgwrite==1.4.1 -sympy==1.6.2 -tensorboard==2.8.0 -tensorboard-data-server==0.6.1 -tensorboard-plugin-wit==1.8.1 -tensorboardX==2.0 -tensorflow==2.3.1 -tensorflow-addons==0.11.2 -tensorflow-estimator==2.3.0 -tensorflow-federated==0.17.0 -tensorflow-model-optimization==0.4.1 -tensorflow-privacy==0.5.2 -termcolor==1.1.0 -terminado==0.13.1 -testpath==0.5.0 -tf-slim==1.1.0 -thinc==7.4.1 -timm==0.4.12 -tinycss2==1.1.1 -tokenizers==0.10.3 -toolz==0.11.2 -torch==1.11.0 -torchtext==0.6.0 -torchvision==0.12.0 -tornado==6.1 -tqdm==4.46.1 -traitlets==5.1.1 -transformers==4.15.0 -typeguard==2.13.3 -typing-extensions==4.1.1 -unification==0.2.2 -urllib3==1.26.8 -virtualenv==20.13.1 -virtualenv-clone==0.5.7 -virtualenvwrapper==4.7.0 -wasabi==0.9.0 -wcwidth==0.2.5 -webencodings==0.5.1 -Werkzeug==2.0.3 -widgetsnbextension==3.4.2 -word2vec==0.11.1 -wrapt==1.13.3 -xlrd==1.2.0 -XlsxWriter==1.4.3 -yellowbrick==1.1 -zipp==3.7.0