first commit

.gitignore

@@ -0,0 +1,2 @@
+.DS_Store
+.ipynb_checkpoints/

app.py

@@ -0,0 +1,24 @@
+import gradio as gr
+import tensorflow as tf
+
+IMG_SIZE = (120, 120)
+
+# load keras model
+model_path = 'banana_model.h5'
+model = tf.keras.models.load_model(model_path)
+
+categories = ('ripe', 'unripe')
+
+def classify_image(img):
+    img_array_expanded_dims = img.reshape((-1, 120, 120, 3))
+    prediction = model.predict(img_array_expanded_dims)
+    prediction_prob = float(tf.nn.sigmoid(prediction))
+    probs = [1-prediction_prob, prediction_prob]
+    return dict(zip(categories, probs))
+
+gr_image = gr.inputs.Image(shape=IMG_SIZE)
+label = gr.outputs.Label()
+examples = ['ripe.jpeg', 'green.jpeg', 'unknown.jpeg']
+
+iface = gr.Interface(fn=classify_image, inputs=gr_image, outputs=label, examples=examples)
+iface.launch(inline=False)

banana_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:572dc383e5c8e16e8d9e294b2b8fff70bef4babdcb056f65881c372d708aab6d
+size 13548640

requirements.txt

@@ -0,0 +1 @@
+tensorflow