init

app.py

@@ -0,0 +1,20 @@
+import streamlit as st
+from transformers import pipeline
+from PIL import Image
+
+pipeline = pipeline(task="image-classification", model="skyau/dog-breed-classifier-vit")
+
+st.title("Dog Breed Classifier?")
+
+file_name = st.file_uploader("Upload a dog image")
+
+if file_name is not None:
+    col1, col2 = st.columns(2)
+
+    image = Image.open(file_name)
+    col1.image(image, use_column_width=True)
+    predictions = pipeline(image)
+
+    col2.header("Probabilities")
+    for p in predictions:
+        col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")

open_ai.yaml

@@ -0,0 +1,125 @@
+name: open_ai
+channels:
+  - conda-forge
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=5.1=1_gnu
+  - asttokens=2.2.1=pyhd8ed1ab_0
+  - backcall=0.2.0=pyh9f0ad1d_0
+  - backports=1.0=pyhd8ed1ab_3
+  - backports.functools_lru_cache=1.6.4=pyhd8ed1ab_0
+  - ca-certificates=2022.12.7=ha878542_0
+  - certifi=2022.12.7=pyhd8ed1ab_0
+  - decorator=5.1.1=pyhd8ed1ab_0
+  - entrypoints=0.4=pyhd8ed1ab_0
+  - executing=1.2.0=pyhd8ed1ab_0
+  - jedi=0.18.2=pyhd8ed1ab_0
+  - jupyter_client=7.0.6=pyhd8ed1ab_0
+  - jupyter_core=5.2.0=py38h578d9bd_0
+  - ld_impl_linux-64=2.38=h1181459_1
+  - libffi=3.4.2=h6a678d5_6
+  - libgcc-ng=11.2.0=h1234567_1
+  - libgomp=11.2.0=h1234567_1
+  - libsodium=1.0.18=h36c2ea0_1
+  - libstdcxx-ng=11.2.0=h1234567_1
+  - matplotlib-inline=0.1.6=pyhd8ed1ab_0
+  - ncurses=6.4=h6a678d5_0
+  - nest-asyncio=1.5.6=pyhd8ed1ab_0
+  - openssl=1.1.1s=h7f8727e_0
+  - packaging=23.0=pyhd8ed1ab_0
+  - parso=0.8.3=pyhd8ed1ab_0
+  - pexpect=4.8.0=pyh1a96a4e_2
+  - pickleshare=0.7.5=py_1003
+  - pip=22.3.1=py38h06a4308_0
+  - prompt-toolkit=3.0.36=pyha770c72_0
+  - ptyprocess=0.7.0=pyhd3deb0d_0
+  - pure_eval=0.2.2=pyhd8ed1ab_0
+  - pygments=2.14.0=pyhd8ed1ab_0
+  - python=3.8.16=h7a1cb2a_2
+  - python-dateutil=2.8.2=pyhd8ed1ab_0
+  - python_abi=3.8=2_cp38
+  - readline=8.2=h5eee18b_0
+  - setuptools=65.6.3=py38h06a4308_0
+  - six=1.16.0=pyh6c4a22f_0
+  - sqlite=3.40.1=h5082296_0
+  - stack_data=0.6.2=pyhd8ed1ab_0
+  - tk=8.6.12=h1ccaba5_0
+  - traitlets=5.9.0=pyhd8ed1ab_0
+  - typing-extensions=4.4.0=hd8ed1ab_0
+  - typing_extensions=4.4.0=pyha770c72_0
+  - wcwidth=0.2.6=pyhd8ed1ab_0
+  - wheel=0.37.1=pyhd3eb1b0_0
+  - xz=5.2.10=h5eee18b_1
+  - zeromq=4.3.4=h9c3ff4c_1
+  - zlib=1.2.13=h5eee18b_0
+  - pip:
+    - aiohttp==3.8.3
+    - aiosignal==1.3.1
+    - altair==4.2.2
+    - async-timeout==4.0.2
+    - attrs==22.2.0
+    - backports-zoneinfo==0.2.1
+    - blinker==1.5
+    - cachetools==4.2.4
+    - charset-normalizer==2.1.1
+    - click==8.1.3
+    - comm==0.1.2
+    - debugpy==1.6.6
+    - frozenlist==1.3.3
+    - gitdb==4.0.10
+    - gitpython==3.1.30
+    - google-api-core==1.34.0
+    - google-auth==1.35.0
+    - google-cloud-core==1.7.3
+    - google-cloud-translate==2.0.1
+    - googleapis-common-protos==1.58.0
+    - grpcio==1.51.1
+    - grpcio-status==1.48.2
+    - idna==3.4
+    - importlib-metadata==6.0.0
+    - importlib-resources==5.10.2
+    - ipykernel==6.21.1
+    - ipython==8.9.0
+    - jinja2==3.1.2
+    - jsonschema==4.17.3
+    - jupyter-client==8.0.2
+    - markdown-it-py==2.1.0
+    - markupsafe==2.1.2
+    - mdurl==0.1.2
+    - multidict==6.0.4
+    - numpy==1.24.2
+    - openai==0.27.0
+    - pandas==1.5.3
+    - pillow==9.4.0
+    - pkgutil-resolve-name==1.3.10
+    - platformdirs==2.6.2
+    - protobuf==3.20.3
+    - psutil==5.9.4
+    - pyarrow==11.0.0
+    - pyasn1==0.4.8
+    - pyasn1-modules==0.2.8
+    - pydeck==0.8.0
+    - pympler==1.0.1
+    - pyrsistent==0.19.3
+    - python-dotenv==0.21.1
+    - pytz==2022.7.1
+    - pytz-deprecation-shim==0.1.0.post0
+    - pyzmq==25.0.0
+    - requests==2.28.2
+    - rich==13.3.1
+    - rsa==4.9
+    - semver==2.13.0
+    - smmap==5.0.0
+    - streamlit==1.17.0
+    - toml==0.10.2
+    - toolz==0.12.0
+    - tornado==6.2
+    - tqdm==4.64.1
+    - tzdata==2022.7
+    - tzlocal==4.2
+    - urllib3==1.26.14
+    - validators==0.20.0
+    - watchdog==2.2.1
+    - yarl==1.8.2
+    - zipp==3.12.1

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch

run.sh

@@ -0,0 +1 @@
+streamlit run app.py

test.ipynb

@@ -0,0 +1,25 @@
+{
+ "cells": [],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "open_ai",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.8.16"
+  },
+  "orig_nbformat": 4
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}