Add application file, pytorch saved models, requirements file

app.py

@@ -0,0 +1,27 @@
+import datasets
+from transformers import AutoFeatureExtractor, AutoModelForImageClassification
+import gradio as gr
+
+
+dataset = datasets.load_dataset("beans")
+
+extractor = AutoFeatureExtractor.from_pretrained("saved_model_files")
+model = AutoModelForImageClassification.from_pretrained("saved_model_files")
+
+labels = dataset['train'].features['labels'].names
+
+def classify(im):
+  features = feature_extractor(im, return_tensors='pt')
+  logits = model(features["pixel_values"])[-1]
+  probability = torch.nn.functional.softmax(logits, dim=-1)
+  probs = probability[0].detach().numpy()
+  confidences = {label: float(probs[i]) for i, label in enumerate(labels)} 
+  return confidences
+
+
+interface = gr.Interface(fn = classify, inputs = "image", outputs = "label",
+                         title = "Plant health classifier",
+                         description = "Classifies plant health"
+                         )
+
+interface.launch()

requirements.txt

@@ -0,0 +1,436 @@
+absl-py==1.2.0
+aeppl==0.0.33
+aesara==2.7.9
+aiohttp==3.8.3
+aiosignal==1.2.0
+alabaster==0.7.12
+albumentations==1.2.1
+altair==4.2.0
+anyio==3.6.1
+appdirs==1.4.4
+arviz==0.12.1
+astor==0.8.1
+astropy==4.3.1
+astunparse==1.6.3
+async-timeout==4.0.2
+asynctest==0.13.0
+atari-py==0.2.9
+atomicwrites==1.4.1
+attrs==22.1.0
+audioread==3.0.0
+autograd==1.5
+Babel==2.10.3
+backcall==0.2.0
+bcrypt==4.0.0
+beautifulsoup4==4.6.3
+bleach==5.0.1
+blis==0.7.8
+bokeh==2.3.3
+branca==0.5.0
+bs4==0.0.1
+CacheControl==0.12.11
+cached-property==1.5.2
+cachetools==4.2.4
+catalogue==2.0.8
+certifi==2022.9.24
+cffi==1.15.1
+cftime==1.6.2
+chardet==3.0.4
+charset-normalizer==2.1.1
+click==7.1.2
+clikit==0.6.2
+cloudpickle==1.5.0
+cmake==3.22.6
+cmdstanpy==1.0.7
+colorcet==3.0.1
+colorlover==0.3.0
+community==1.0.0b1
+confection==0.0.2
+cons==0.4.5
+contextlib2==0.5.5
+convertdate==2.4.0
+crashtest==0.3.1
+crcmod==1.7
+cryptography==38.0.1
+cufflinks==0.17.3
+cupy-cuda11x==11.0.0
+cvxopt==1.3.0
+cvxpy==1.2.1
+cycler==0.11.0
+cymem==2.0.6
+Cython==0.29.32
+daft==0.0.4
+dask==2022.2.0
+datascience==0.17.5
+datasets==2.5.2
+debugpy==1.0.0
+decorator==4.4.2
+defusedxml==0.7.1
+descartes==1.1.0
+dill==0.3.5.1
+distributed==2022.2.0
+dlib==19.24.0
+dm-tree==0.1.7
+docutils==0.17.1
+dopamine-rl==1.0.5
+earthengine-api==0.1.326
+easydict==1.10
+ecos==2.0.10
+editdistance==0.5.3
+en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.4.0/en_core_web_sm-3.4.0-py3-none-any.whl
+entrypoints==0.4
+ephem==4.1.3
+et-xmlfile==1.1.0
+etils==0.8.0
+etuples==0.3.8
+evaluate==0.2.2
+fa2==0.3.5
+fastai==2.7.9
+fastapi==0.85.0
+fastcore==1.5.27
+fastdownload==0.0.7
+fastdtw==0.3.4
+fastjsonschema==2.16.2
+fastprogress==1.0.3
+fastrlock==0.8
+feather-format==0.4.1
+ffmpy==0.3.0
+filelock==3.8.0
+firebase-admin==4.4.0
+fix-yahoo-finance==0.0.22
+Flask==1.1.4
+flatbuffers==22.9.24
+folium==0.12.1.post1
+frozenlist==1.3.1
+fsspec==2022.8.2
+future==0.16.0
+gast==0.5.3
+GDAL==2.2.2
+gdown==4.4.0
+gensim==3.6.0
+geographiclib==1.52
+geopy==1.17.0
+gin-config==0.5.0
+glob2==0.7
+google==2.0.3
+google-api-core==1.31.6
+google-api-python-client==1.12.11
+google-auth==1.35.0
+google-auth-httplib2==0.0.4
+google-auth-oauthlib==0.4.6
+google-cloud-bigquery==1.21.0
+google-cloud-bigquery-storage==1.1.2
+google-cloud-core==1.0.3
+google-cloud-datastore==1.8.0
+google-cloud-firestore==1.7.0
+google-cloud-language==1.2.0
+google-cloud-storage==1.18.1
+google-cloud-translate==1.5.0
+google-colab @ file:///colabtools/dist/google-colab-1.0.0.tar.gz
+google-pasta==0.2.0
+google-resumable-media==0.4.1
+googleapis-common-protos==1.56.4
+googledrivedownloader==0.4
+gradio==3.4.1
+graphviz==0.10.1
+greenlet==1.1.3
+grpcio==1.49.1
+gspread==3.4.2
+gspread-dataframe==3.0.8
+gym==0.25.2
+gym-notices==0.0.8
+h11==0.12.0
+h5py==3.1.0
+HeapDict==1.0.1
+hijri-converter==2.2.4
+holidays==0.16
+holoviews==1.14.9
+html5lib==1.0.1
+httpcore==0.15.0
+httpimport==0.5.18
+httplib2==0.17.4
+httplib2shim==0.0.3
+httpstan==4.6.1
+httpx==0.23.0
+huggingface-hub==0.10.0
+humanize==0.5.1
+hyperopt==0.1.2
+idna==2.10
+imageio==2.9.0
+imagesize==1.4.1
+imbalanced-learn==0.8.1
+imblearn==0.0
+imgaug==0.4.0
+importlib-metadata==5.0.0
+importlib-resources==5.9.0
+imutils==0.5.4
+inflect==2.1.0
+intel-openmp==2022.2.0
+intervaltree==2.1.0
+ipykernel==5.3.4
+ipython==7.9.0
+ipython-genutils==0.2.0
+ipython-sql==0.3.9
+ipywidgets==7.7.1
+itsdangerous==1.1.0
+jax==0.3.21
+jaxlib @ https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.20+cuda11.cudnn805-cp37-cp37m-manylinux2014_x86_64.whl
+jieba==0.42.1
+Jinja2==2.11.3
+joblib==1.2.0
+jpeg4py==0.1.4
+jsonschema==4.3.3
+jupyter-client==6.1.12
+jupyter-console==6.1.0
+jupyter-core==4.11.1
+jupyterlab-widgets==3.0.3
+kaggle==1.5.12
+kapre==0.3.7
+keras==2.8.0
+Keras-Preprocessing==1.1.2
+keras-vis==0.4.1
+kiwisolver==1.4.4
+korean-lunar-calendar==0.3.1
+langcodes==3.3.0
+libclang==14.0.6
+librosa==0.8.1
+lightgbm==2.2.3
+linkify-it-py==1.0.3
+llvmlite==0.39.1
+lmdb==0.99
+locket==1.0.0
+logical-unification==0.4.5
+LunarCalendar==0.0.9
+lxml==4.9.1
+Markdown==3.4.1
+markdown-it-py==2.1.0
+MarkupSafe==2.0.1
+marshmallow==3.18.0
+matplotlib==3.2.2
+matplotlib-venn==0.11.7
+mdit-py-plugins==0.3.1
+mdurl==0.1.2
+miniKanren==1.0.3
+missingno==0.5.1
+mistune==0.8.4
+mizani==0.7.3
+mkl==2019.0
+mlxtend==0.14.0
+more-itertools==8.14.0
+moviepy==0.2.3.5
+mpmath==1.2.1
+msgpack==1.0.4
+multidict==6.0.2
+multipledispatch==0.6.0
+multiprocess==0.70.13
+multitasking==0.0.11
+murmurhash==1.0.8
+music21==5.5.0
+natsort==5.5.0
+nbconvert==5.6.1
+nbformat==5.6.1
+netCDF4==1.6.1
+networkx==2.6.3
+nibabel==3.0.2
+nltk==3.7
+notebook==5.3.1
+numba==0.56.2
+numexpr==2.8.3
+numpy==1.21.6
+oauth2client==4.1.3
+oauthlib==3.2.1
+okgrade==0.4.3
+opencv-contrib-python==4.6.0.66
+opencv-python==4.6.0.66
+opencv-python-headless==4.6.0.66
+openpyxl==3.0.10
+opt-einsum==3.3.0
+orjson==3.8.0
+osqp==0.6.2.post0
+packaging==21.3
+palettable==3.3.0
+pandas==1.3.5
+pandas-datareader==0.9.0
+pandas-gbq==0.13.3
+pandas-profiling==1.4.1
+pandocfilters==1.5.0
+panel==0.12.1
+param==1.12.2
+paramiko==2.11.0
+parso==0.8.3
+partd==1.3.0
+pastel==0.2.1
+pathlib==1.0.1
+pathy==0.6.2
+patsy==0.5.2
+pep517==0.13.0
+pexpect==4.8.0
+pickleshare==0.7.5
+Pillow==7.1.2
+pip-tools==6.2.0
+plotly==5.5.0
+plotnine==0.8.0
+pluggy==0.7.1
+pooch==1.6.0
+portpicker==1.3.9
+prefetch-generator==1.0.1
+preshed==3.0.7
+prettytable==3.4.1
+progressbar2==3.38.0
+promise==2.3
+prompt-toolkit==2.0.10
+prophet==1.1.1
+protobuf==3.17.3
+psutil==5.4.8
+psycopg2==2.9.3
+ptyprocess==0.7.0
+py==1.11.0
+pyarrow==6.0.1
+pyasn1==0.4.8
+pyasn1-modules==0.2.8
+pycocotools==2.0.5
+pycparser==2.21
+pycryptodome==3.15.0
+pyct==0.4.8
+pydantic==1.9.2
+pydata-google-auth==1.4.0
+pydot==1.3.0
+pydot-ng==2.0.0
+pydotplus==2.0.2
+PyDrive==1.3.1
+pydub==0.25.1
+pyemd==0.5.1
+pyerfa==2.0.0.1
+Pygments==2.6.1
+pygobject==3.26.1
+pylev==1.4.0
+pymc==4.1.4
+PyMeeus==0.5.11
+pymongo==4.2.0
+pymystem3==0.2.0
+PyNaCl==1.5.0
+PyOpenGL==3.1.6
+pyparsing==3.0.9
+pyrsistent==0.18.1
+pysimdjson==3.2.0
+pysndfile==1.3.8
+PySocks==1.7.1
+pystan==3.3.0
+pytest==3.6.4
+python-apt==0.0.0
+python-chess==0.23.11
+python-dateutil==2.8.2
+python-louvain==0.16
+python-multipart==0.0.5
+python-slugify==6.1.2
+python-utils==3.3.3
+pytz==2022.4
+pyviz-comms==2.2.1
+PyWavelets==1.3.0
+PyYAML==6.0
+pyzmq==23.2.1
+qdldl==0.1.5.post2
+qudida==0.0.4
+regex==2022.6.2
+requests==2.23.0
+requests-oauthlib==1.3.1
+resampy==0.4.2
+responses==0.18.0
+rfc3986==1.5.0
+rpy2==3.4.5
+rsa==4.9
+scikit-image==0.18.3
+scikit-learn==1.0.2
+scipy==1.7.3
+screen-resolution-extra==0.0.0
+scs==3.2.0
+seaborn==0.11.2
+Send2Trash==1.8.0
+setuptools-git==1.2
+Shapely==1.8.4
+six==1.15.0
+sklearn-pandas==1.8.0
+smart-open==5.2.1
+sniffio==1.3.0
+snowballstemmer==2.2.0
+sortedcontainers==2.4.0
+soundfile==0.11.0
+spacy==3.4.1
+spacy-legacy==3.0.10
+spacy-loggers==1.0.3
+Sphinx==1.8.6
+sphinxcontrib-serializinghtml==1.1.5
+sphinxcontrib-websupport==1.2.4
+SQLAlchemy==1.4.41
+sqlparse==0.4.3
+srsly==2.4.4
+starlette==0.20.4
+statsmodels==0.12.2
+sympy==1.7.1
+tables==3.7.0
+tabulate==0.8.10
+tblib==1.7.0
+tenacity==8.1.0
+tensorboard==2.8.0
+tensorboard-data-server==0.6.1
+tensorboard-plugin-wit==1.8.1
+tensorflow==2.8.2+zzzcolab20220929150707
+tensorflow-datasets==4.6.0
+tensorflow-estimator==2.8.0
+tensorflow-gcs-config==2.8.0
+tensorflow-hub==0.12.0
+tensorflow-io-gcs-filesystem==0.27.0
+tensorflow-metadata==1.10.0
+tensorflow-probability==0.16.0
+termcolor==2.0.1
+terminado==0.13.3
+testpath==0.6.0
+text-unidecode==1.3
+textblob==0.15.3
+thinc==8.1.2
+threadpoolctl==3.1.0
+tifffile==2021.11.2
+tokenizers==0.12.1
+toml==0.10.2
+tomli==2.0.1
+toolz==0.12.0
+torch @ https://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp37-cp37m-linux_x86_64.whl
+torchaudio @ https://download.pytorch.org/whl/cu113/torchaudio-0.12.1%2Bcu113-cp37-cp37m-linux_x86_64.whl
+torchsummary==1.5.1
+torchtext==0.13.1
+torchvision @ https://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp37-cp37m-linux_x86_64.whl
+tornado==5.1.1
+tqdm==4.64.1
+traitlets==5.1.1
+transformers==4.22.2
+tweepy==3.10.0
+typeguard==2.7.1
+typer==0.4.2
+typing-extensions==4.1.1
+tzlocal==1.5.1
+uc-micro-py==1.0.1
+ujson==5.5.0
+uritemplate==3.0.1
+urllib3==1.25.11
+uvicorn==0.18.3
+vega-datasets==0.9.0
+wasabi==0.10.1
+wcwidth==0.2.5
+webargs==8.2.0
+webencodings==0.5.1
+websockets==10.3
+Werkzeug==1.0.1
+widgetsnbextension==3.6.1
+wordcloud==1.8.2.2
+wrapt==1.14.1
+xarray==0.20.2
+xarray-einstats==0.2.2
+xgboost==0.90
+xkit==0.0.0
+xlrd==1.1.0
+xlwt==1.3.0
+xxhash==3.0.0
+yarl==1.8.1
+yellowbrick==1.5
+zict==2.2.0
+zipp==3.8.1

saved_model_files/config.json

@@ -0,0 +1,34 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "angular_leaf_spot",
+    "1": "bean_rust",
+    "2": "healthy"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "angular_leaf_spot": "0",
+    "bean_rust": "1",
+    "healthy": "2"
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.22.2"
+}

saved_model_files/preprocessor_config.json

@@ -0,0 +1,17 @@
+{
+  "do_normalize": true,
+  "do_resize": true,
+  "feature_extractor_type": "ViTFeatureExtractor",
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "size": 224
+}

saved_model_files/pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1544bd049155d33a8bd6e3e6f90fe34197403c925bd38c8ace5091479952080d
+size 343270065

saved_model_files/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ee592c0d94564ee3bffdde318a595c528aaf89d2a6104f078687478119c9912
+size 3375