git restart

.gitattributes

@@ -0,0 +1,35 @@
+*.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
+*.ckpt 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
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz 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
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl 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
+*.safetensors 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
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+model4app.onnx filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,13 @@
+---
+title: Birds Israel
+emoji: 🐢
+colorFrom: yellow
+colorTo: green
+sdk: streamlit
+sdk_version: 1.15.2
+app_file: app.py
+pinned: false
+license: cc-by-nc-nd-4.0
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,22 @@
+import streamlit as st
+from PIL import Image
+
+from bird_app import BirdApp
+
+bird_app = BirdApp()
+
+st.title("A Classifier for Israel's birds")
+
+file_name = st.file_uploader("Upload an image of a bird")
+
+if file_name is not None:
+    col1, col2 = st.columns(2)
+    image = Image.open(file_name)
+
+    predictions = bird_app.predict(file_name)
+
+    col1.image(image, use_column_width=True)
+
+    col2.header(f"Predicted species: {predictions['class_name']}")
+    # for p in predictions:
+    #   col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")

bird_app.py

@@ -0,0 +1,30 @@
+import json
+import numpy as np
+import onnxruntime
+from PIL import Image
+
+
+class BirdApp:
+    def __init__(self):
+        self.onnx_session = onnxruntime.InferenceSession("model4app2.onnx")
+        self.img_class_map = get_img_class_map()
+
+    def predict(self, x):
+        input_tensor = transform_image(x)
+        onnx_inputs = {self.onnx_session.get_inputs()[0].name: input_tensor}
+        img_label = self.onnx_session.run(None, onnx_inputs)[0].argmax()
+        return {'class_id': int(img_label), 'class_name': self.img_class_map[str(img_label)]}
+
+
+def transform_image(infile) -> np.array:
+    image = (Image
+             .open(infile)
+             .resize((224, 224))
+             )
+    return np.expand_dims(np.array(image, dtype=np.float32), 0).transpose([0, 3, 1, 2])
+
+
+def get_img_class_map():
+    with open('index_to_name.json') as f:
+        img_class_map = json.load(f)
+    return img_class_map

index_to_name.json

@@ -0,0 +1 @@
+{"0": "blackbird", "1": "blackcap", "2": "bulbul", "3": "crow", "4": "jay", "5": "kestrel", "6": "kingfisher", "7": "laughing_dove", "8": "myna", "9": "parakeet", "10": "pigeon", "11": "prinia", "12": "sparrow", "13": "sunbird", "14": "swallow", "15": "swift", "16": "tit", "17": "wagtail"}

model4app.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f388784ed04cc4bb2c88300e9f42e1d038d69bf82a06ee1e684819c08b7610c
+size 94167203

requirements.txt

@@ -0,0 +1,6 @@
+Pillow~=9.0.0
+onnxruntime
+numpy==1.22.3
+jsonschema~=4.4.0
+Flask==2.1.2
+gunicorn==20.0.4