Add processing

app.py

@@ -1,3 +1,29 @@
 import gradio as gr
+from huggingface_hub import from_pretrained_keras
+from tensorflow.keras.preprocessing.image import array_to_img, img_to_array
 
-gr.Interface.load("models/jsolow/grubguesser").launch()
+from .labels import CLASS_LABELS
+
+
+model = from_pretrained_keras("jsolow/grubguesser")
+
+
+def _preprocess(imgage_input):
+    img = array_to_img(imgage_input, scale=False)
+    img = img.resize((224, 224))
+    img = img_to_array(img)
+    return img / 255.0
+
+
+def predict(image):
+    image_array = _preprocess(image)
+    predictions = model.predict(image_array)
+    return {k:v for k, v in zip(CLASS_LABELS, predictions)}
+
+
+gr.Interface(
+    predict,
+    inputs=gr.inputs.Image(label="Upload food image", type="filepath"),
+    outputs=gr.outputs.Label(num_top_classes=10),
+    title="Grubguesser - What is this food?",
+).launch()