Update Model

README.md

@@ -1,6 +1,6 @@
 ---
-title: Coffee Grader
-emoji: 🐢
+title: Coffee Bean Grader
+emoji: ☕
 colorFrom: blue
 colorTo: blue
 sdk: gradio

app.py

@@ -0,0 +1,15 @@
+import tensorflow
+from tensorflow import keras
+import gradio as gr
+
+model = keras.models.load_model('mymodel.h5')
+potato_classes = ['Grade_3','Grade_2','Grade_1']
+
+def predict_input_image(img):
+  img_3d=img.reshape(-1,256,256,3)
+  prediction=model.predict(img_3d)[0]
+  return {potato_classes[i]: float(prediction[i]) for i in range(3)}
+
+image = gr.inputs.Image(shape=(256,256))
+label = gr.outputs.Label(num_top_classes=3)
+gr.Interface(fn=predict_input_image, inputs=image, outputs=label,interpretation='default').launch()

mymodel.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:99c620a6eead6ed33d6cf6b6ab1d0494d2735c59cd2bbefa809527234f3798f6
+size 34118040

requirements.txt

@@ -0,0 +1,2 @@
+numpy==1.21.6
+tensorflow==2.9.2