add: gradio interface

app.py

@@ -0,0 +1,13 @@
+import gradio as gr
+import os
+from functions import *
+
+examples_dir = 'examples'
+title = "Birds Classification - ResNet34 PyTorch"
+examples = [os.path.join(examples_dir, i) for i in os.listdir('examples')]
+
+interface = gr.Interface(fn=predict, inputs=gr.Image(type= 'numpy', shape=(64, 64)).style(height= 256),
+            outputs= gr.Label(num_top_classes= 5), cache_examples= False,
+            examples= examples, title= title)
+
+interface.launch()

functions.py

@@ -0,0 +1,47 @@
+import numpy as np
+import json
+import onnxruntime as rt
+
+model_path = 'models/model.onnx'
+idx_to_class = 'models/idx_to_class.json'
+
+normalise_means = [0.4914, 0.4822, 0.4465]
+normalise_stds = [0.2023, 0.1994, 0.2010]
+
+def normalise_image(image):
+    image = image.copy()
+    for i in range(3):
+        image[:, i, :, :] = (image[:, i, :, :] - normalise_means[i]) / normalise_stds[i]
+    return image
+
+def load_class_names():
+    with open(idx_to_class, 'r') as f:
+        class_names = json.load(f)
+    return class_names
+
+def predict(inp_image):
+    
+    class_names = load_class_names()
+
+    image = inp_image
+    image = image.transpose((2, 0, 1))
+
+    image = image / 255.0
+    image = np.expand_dims(image, axis=0)
+    image = normalise_image(image)
+    image = image.astype(np.float32)
+
+    sess = rt.InferenceSession(model_path)
+
+    input_name = sess.get_inputs()[0].name
+    output_name = sess.get_outputs()[0].name
+
+    output = sess.run([output_name], {input_name: image})[0]
+    prob = np.exp(output) / np.sum(np.exp(output), axis=1, keepdims=True)
+
+    top5 = np.argsort(prob[0])[-5:][::-1]
+
+    class_probs = {class_names[str(i)]: float(prob[0][i]) for i in top5}
+    print(class_probs)
+
+    return class_probs

requirements.txt

@@ -0,0 +1,12 @@
+gradio==3.16.1
+json5==0.9.10
+matplotlib==3.6.2
+matplotlib-inline==0.1.6
+numpy==1.23.5
+onnx==1.13.0
+onnxruntime==1.13.1
+pandas==1.5.2
+Pillow==9.3.0
+torch==1.13.1
+torchvision==0.14.1
+tqdm==4.64.1