Duplicate from mikachou/dog-breed-identification

Co-authored-by: Mika Chou <mikachou@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,27 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.python-version

@@ -0,0 +1 @@
+3.7.13

README.md

@@ -0,0 +1,14 @@
+---
+title: Dog Breed Identification
+emoji: 🐕
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: 3.0.20
+app_file: app.py
+pinned: false
+python_version: 3.7.13
+duplicated_from: mikachou/dog-breed-identification
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,45 @@
+import json
+import tensorflow as tf
+from huggingface_hub import hf_hub_download
+import gradio as gr
+
+tf_model = hf_hub_download(repo_id='mikachou/dog-breed-classifier', filename='tf_model.h5')
+config_json = hf_hub_download(repo_id='mikachou/dog-breed-classifier', filename='config.json')
+
+model = tf.keras.models.load_model(tf_model)
+print(model.summary())
+
+with open(config_json) as f:
+    config = json.load(f)
+
+dogs_breeds = list(config['id2label'].values())
+
+def predict(filepath):
+    img = tf.io.read_file(filepath)
+    tensor = tf.io.decode_image(img, channels=3, dtype=tf.dtypes.float32)
+    tensor = tf.image.resize(tensor, [299, 299])
+    input_tensor = tf.expand_dims(tensor, axis=0)
+
+    output = model.predict(input_tensor)
+
+    confidences = { dogs_breeds[i]: float(output[0][i]) for i in range(120) }
+
+    return confidences
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(label='photo', type='filepath'),
+    outputs=gr.Label(label="Predicted breed", num_top_classes=3),
+    examples=[
+        'imgs/beethoven.jpg',
+        'imgs/belle.png',
+        'imgs/belmondo.jpg',
+        'imgs/dorothy.jpg',
+        'imgs/lassie.jpg',
+        'imgs/rintintin.jpg'
+    ],
+    title="Dog breed identification",
+    description="The model was trained with [Stanford Dogs Dataset](http://vision.stanford.edu/aditya86/ImageNetDogs/) using tensorflow/keras on a fine-tuned pre-trained InceptionResNetV2 model",
+    article="You could also drag/drop other examples from [this page](https://www.rdasia.com/pets/can-you-guess-dog-breed-based-its-puppy-picture)")
+
+demo.launch()

requirements.txt

@@ -0,0 +1,93 @@
+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==2022.6.15
+cffi==1.15.0
+charset-normalizer==2.0.12
+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.8.0
+google-auth-oauthlib==0.4.6
+google-pasta==0.2.0
+gradio==3.0.20
+grpcio==1.47.0
+h11==0.13.0
+h5py==3.7.0
+huggingface-hub==0.8.1
+idna==3.3
+importlib-metadata==4.12.0
+Jinja2==3.1.2
+keras==2.8.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
+opt-einsum==3.3.0
+orjson==3.7.5
+packaging==21.3
+pandas==1.3.5
+paramiko==2.11.0
+Pillow==9.1.1
+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
+requests==2.28.0
+requests-oauthlib==1.3.1
+rsa==4.8
+six==1.16.0
+sniffio==1.2.0
+starlette==0.19.1
+tensorboard==2.8.0
+tensorboard-data-server==0.6.1
+tensorboard-plugin-wit==1.8.1
+tensorflow==2.8.2
+tensorflow-estimator==2.8.0
+tensorflow-io-gcs-filesystem==0.26.0
+termcolor==1.1.0
+tqdm==4.64.0
+typing_extensions==4.2.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