Initial commit

app.py

@@ -0,0 +1,38 @@
+import gradio as gr
+from fastai.vision.all import load_learner
+from pathlib import Path
+import glob
+
+classes_file = Path('classes.txt')
+if not classes_file.exists():
+    raise FileNotFoundError(f"{classes_file} not found")
+classes = classes_file.read_text().splitlines()
+
+model_path = Path('model.pkl')
+if not model_path.exists():
+    raise FileNotFoundError(f"{model_path} not found")
+learn = load_learner(model_path)
+
+sample_folder = Path('samples')
+if sample_folder.exists():
+    sample_images = sorted(glob.glob(str(sample_folder / '*')))
+    examples = [[img] for img in sample_images]
+else:
+    examples = []
+
+def predict(img):
+    # img: PIL image
+    pred, idx, probs = learn.predict(img)
+    return {classes[i]: float(probs[i]) for i in range(len(classes))}
+
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type='pil'),
+    outputs=gr.Label(num_top_classes=5),
+    examples=examples,
+    title="Food-101 Classifier",
+    description="Upload an image of food or choose from examples to get predictions."
+)
+
+if __name__ == "__main__":
+    iface.launch()

classes.txt

@@ -0,0 +1,101 @@
+apple_pie
+baby_back_ribs
+baklava
+beef_carpaccio
+beef_tartare
+beet_salad
+beignets
+bibimbap
+bread_pudding
+breakfast_burrito
+bruschetta
+caesar_salad
+cannoli
+caprese_salad
+carrot_cake
+ceviche
+cheesecake
+cheese_plate
+chicken_curry
+chicken_quesadilla
+chicken_wings
+chocolate_cake
+chocolate_mousse
+churros
+clam_chowder
+club_sandwich
+crab_cakes
+creme_brulee
+croque_madame
+cup_cakes
+deviled_eggs
+donuts
+dumplings
+edamame
+eggs_benedict
+escargots
+falafel
+filet_mignon
+fish_and_chips
+foie_gras
+french_fries
+french_onion_soup
+french_toast
+fried_calamari
+fried_rice
+frozen_yogurt
+garlic_bread
+gnocchi
+greek_salad
+grilled_cheese_sandwich
+grilled_salmon
+guacamole
+gyoza
+hamburger
+hot_and_sour_soup
+hot_dog
+huevos_rancheros
+hummus
+ice_cream
+lasagna
+lobster_bisque
+lobster_roll_sandwich
+macaroni_and_cheese
+macarons
+miso_soup
+mussels
+nachos
+omelette
+onion_rings
+oysters
+pad_thai
+paella
+pancakes
+panna_cotta
+peking_duck
+pho
+pizza
+pork_chop
+poutine
+prime_rib
+pulled_pork_sandwich
+ramen
+ravioli
+red_velvet_cake
+risotto
+samosa
+sashimi
+scallops
+seaweed_salad
+shrimp_and_grits
+spaghetti_bolognese
+spaghetti_carbonara
+spring_rolls
+steak
+strawberry_shortcake
+sushi
+tacos
+takoyaki
+tiramisu
+tuna_tartare
+waffles

model.pkl

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