changed requirements

app.py

@@ -186,7 +186,7 @@ def predict_animal_mask(im):
     bboxes = outputs.pred_boxes
     masks = outputs.pred_masks # torch.Size([1, 100, 200, 200])
 
-    # postprocess the image
+    # postprocess the mask (numpy arrays)
     label_per_pixel = torch.argmax(masks.squeeze(),dim=0).detach().numpy()
     color_mask = np.zeros(image.size+(3,))
     for lbl, color in enumerate(ade_palette()):

requirements.txt

@@ -1,104 +1,6 @@
-absl-py==1.1.0
-aiohttp==3.8.1
-aiosignal==1.2.0
-analytics-python==1.4.0
-anyio==3.6.1
-astunparse==1.6.3
-async-timeout==4.0.2
-asynctest==0.13.0
-attrs==21.4.0
-backoff==1.10.0
-bcrypt==3.2.2
-cachetools==5.2.0
-certifi @ file:///private/var/folders/sy/f16zz6x50xz3113nwtb9bvq00000gp/T/abs_83242e7e-f82d-4a71-8ef2-9d71d212d249gu_wxmeq/croots/recipe/certifi_1655968827803/work/certifi
-cffi==1.15.1
-charset-normalizer==2.1.0
-click==8.1.3
-cryptography==37.0.2
-cycler==0.11.0
-fastapi==0.78.0
-ffmpy==0.3.0
-filelock==3.7.1
-flatbuffers==1.12
-fonttools==4.33.3
-frozenlist==1.3.0
-fsspec==2022.5.0
-gast==0.4.0
-google-auth==2.9.0
-google-auth-oauthlib==0.4.6
-google-pasta==0.2.0
-gradio==3.0.24
-grpcio==1.47.0
-h11==0.12.0
-h5py==3.7.0
-httpcore==0.15.0
-httpx==0.23.0
-huggingface-hub==0.8.1
-idna==3.3
-importlib-metadata==4.12.0
-Jinja2==3.1.2
-keras==2.9.0
-Keras-Preprocessing==1.1.2
-kiwisolver==1.4.3
-libclang==14.0.1
-linkify-it-py==1.0.3
-Markdown==3.3.7
-markdown-it-py==2.1.0
-MarkupSafe==2.1.1
-matplotlib==3.5.2
-mdit-py-plugins==0.3.0
-mdurl==0.1.1
-monotonic==1.6
-multidict==6.0.2
-numpy==1.21.6
-oauthlib==3.2.0
-olefile==0.46
-opt-einsum==3.3.0
-orjson==3.7.6
-packaging==21.3
-pandas==1.3.5
-paramiko==2.11.0
-Pillow @ file:///Users/runner/miniforge3/conda-bld/pillow_1602493105252/work
-protobuf==3.19.4
-pyasn1==0.4.8
-pyasn1-modules==0.2.8
-pycparser==2.21
-pycryptodome==3.15.0
-pydantic==1.9.1
-pydub==0.25.1
-PyNaCl==1.5.0
-pyparsing==3.0.9
-python-dateutil==2.8.2
-python-multipart==0.0.5
-pytz==2022.1
-PyYAML==6.0
-regex==2022.6.2
-requests==2.28.1
-requests-oauthlib==1.3.1
-rfc3986==1.5.0
-rsa==4.8
-six==1.16.0
-sniffio==1.2.0
-starlette==0.19.1
-tensorboard==2.9.1
-tensorboard-data-server==0.6.1
-tensorboard-plugin-wit==1.8.1
-tensorflow==2.9.1
-tensorflow-estimator==2.9.0
-tensorflow-io-gcs-filesystem==0.26.0
-termcolor==1.1.0
-timm==0.5.4
-tokenizers==0.12.1
-torch==1.12.0
-torchaudio==0.12.0
-torchvision==0.13.0
-tqdm==4.64.0
-transformers==4.20.1
-typing_extensions==4.3.0
-uc-micro-py==1.0.1
-urllib3==1.26.9
-uvicorn==0.18.2
-Werkzeug==2.1.2
-wrapt==1.14.1
-yarl==1.7.2
-zipp==3.8.0
+transformers
+Pillow
+gradio
+numpy
+torch
+torchvision

requirements_OLD.txt

@@ -1,6 +0,0 @@
-transformers
-Pillow
-gradio
-numpy
-torch
-torchvision