Add application file

app.py

@@ -0,0 +1,271 @@
+import gradio as gr
+from fastai.vision.all import load_learner
+from pathlib import Path
+
+
+
+food_labels = (
+    'adhirasam',
+    'aloo_gobi',
+    'aloo_matar',
+    'aloo_methi',
+    'aloo_shimla_mirch',
+    'aloo_tikki',
+    'anarsa',
+    'apple_pie',
+    'ariselu',
+    'baby_back_ribs',
+    'Baden_Brägele_with_Bibliskäs',
+    'baklava',
+    'bandar_laddu',
+    'basundi',
+    'beef_carpaccio',
+    'beef_tartare',
+    'beet_salad',
+    'beignets',
+    'bhatura',
+    'bhindi_masala',
+    'Bibimbap_food',
+    'biryani',
+    'Blanquette_de_Veau',
+    'Boeuf_Bourguignon',
+    'boondi',
+    'Bossam_food',
+    'Bouillabaisse',
+    'Braised_Beef_Rheinland_style',
+    'Braunkohl_and_Bregenwurst',
+    'bread_pudding',
+    'breakfast_burrito',
+    'bruschetta',
+    'Budae_Jjigae_food',
+    'Bulgogi_food',
+    'butter_chicken',
+    'Butter_Noodles',
+    'caesar_salad',
+    'cannoli',
+    'caprese_salad',
+    'carrot_cake',
+    'Cassoulet',
+    'ceviche',
+    'chak_hao_kheer',
+    'cham_cham',
+    'chana_masala',
+    'chapati',
+    'cheesecake',
+    'cheese_plate',
+    'chhena_kheeri',
+    'chicken_quesadilla',
+    'chicken_razala',
+    'chicken_tikka',
+    'chicken_tikka_masala',
+    'chicken_wings',
+    'chikki',
+    'chocolate_cake',
+    'chocolate_mousse',
+    'Chuncheon_Dakgalbi_food',
+    'churros',
+    'clam_chowder',
+    'club_sandwich',
+    'Coquilles_Saint_Jacques',
+    'Coq_au_Vin',
+    'crab_cakes',
+    'creme_brulee',
+    'croque_madame',
+    'Crème_brûlée',
+    'Crêpes',
+    'cup_cakes',
+    'daal_baati_churma',
+    'daal_puri',
+    'dal_makhani',
+    'dal_tadka',
+    'deviled_eggs',
+    'dharwad_pedha',
+    'donuts',
+    'doodhpak',
+    'double_ka_meetha',
+    'Dumplings',
+    'dum_aloo',
+    'Döner_with_Curried_Sausage',
+    'edamame',
+    'eggs_benedict',
+    'Escargots',
+    'falafel',
+    'filet_mignon',
+    'fish_and_chips',
+    'foie_gras',
+    'Fondue_Savoyarde',
+    'Frankfurter_Green_Sauce',
+    'french_fries',
+    'french_onion_soup',
+    'french_toast',
+    'fried_calamari',
+    'fried_rice',
+    'frozen_yogurt',
+    'gajar_ka_halwa',
+    'garlic_bread',
+    'gavvalu',
+    'German_Bread_Rolls',
+    'ghevar',
+    'Gimbap_food',
+    'gnocchi',
+    'greek_salad',
+    'grilled_cheese_sandwich',
+    'grilled_salmon',
+    'guacamole',
+    'gulab_jamun',
+    'gyoza',
+    'Gyudon',
+    'hamburger',
+    'Hamburger_Labskaus',
+    'Hamburger_Pannfisch',
+    'hot_and_sour_soup',
+    'hot_dog',
+    'huevos_rancheros',
+    'hummus',
+    'ice_cream',
+    'imarti',
+    'jalebi',
+    'kachori',
+    'kadai_paneer',
+    'kadhi_pakoda',
+    'kajjikaya',
+    'kakinada_khaja',
+    'kalakand',
+    'Kalguksu_food',
+    'karela_bharta',
+    'Kare_raisu',
+    'Katsudon',
+    'Kimchi_Jjigae_food',
+    'kofta',
+    'kuzhi_paniyaram',
+    'Königsberger_Meatballs',
+    'lasagna',
+    'lassi',
+    'ledikeni',
+    'litti_chokha',
+    'lobster_bisque',
+    'lobster_roll_sandwich',
+    'lyangcha',
+    'maach_jhol',
+    'macaroni_and_cheese',
+    'macarons',
+    'makki_di_roti_sarson_da_saag',
+    'malapua',
+    'Manduguk_food',
+    'misi_roti',
+    'Miso_soup',
+    'misti_doi',
+    'modak',
+    'Moules_Marinières',
+    'mussels',
+    'mysore_pak',
+    'naan',
+    'nachos',
+    'navrattan_korma',
+    'Ojingeo_Bokkeum_food',
+    'Okonomiyaki',
+    'omelette',
+    'Omurice',
+    'Onigiri',
+    'onion_rings',
+    'oysters',
+    'pad_thai',
+    'paella',
+    'Pajeon_food',
+    'palak_paneer',
+    'pancakes',
+    'paneer_butter_masala',
+    'panna_cotta',
+    'peking_duck',
+    'phirni',
+    'pho',
+    'Pissaladière',
+    'pithe',
+    'pizza',
+    'poha',
+    'poornalu',
+    'pootharekulu',
+    'pork_chop',
+    'Pork_Knuckle_with_Sauerkraut',
+    'Potato_Fritters_or_Potato_Pancakes_Kartoffelpuffer_',
+    'Potato_Salad',
+    'poutine',
+    'Pretzen_from_Bavaria',
+    'prime_rib',
+    'pulled_pork_sandwich',
+    'Pumpernickel_from_Lower_Saxony',
+    'qubani_ka_meetha',
+    'rabri',
+    'Ramen',
+    'rasgulla',
+    'ras_malai',
+    'ravioli',
+    'red_velvet_cake',
+    'risotto',
+    'Salade_niçoise',
+    'Samgyetang_food',
+    'samosa',
+    'sandesh',
+    'sashimi',
+    'scallops',
+    'seaweed_salad',
+    'Shabu_Shabu',
+    'shankarpali',
+    'sheera',
+    'sheer_korma',
+    'shrikhand',
+    'shrimp_and_grits',
+    'Soba',
+    'sohan_halwa',
+    'sohan_papdi',
+    'Soufflé',
+    'Soupe_à_L_oignon',
+    'spaghetti_bolognese',
+    'spaghetti_carbonara',
+    'spring_rolls',
+    'steak',
+    'Steak_Tartare',
+    'strawberry_shortcake',
+    'Sundae_food',
+    'sushi',
+    'Sushi_Sashimi',
+    'sutar_feni',
+    'tacos',
+    'takoyaki',
+    'Tarte_Flambée_with_a_French_Influence',
+    'Tarte_Tatin',
+    'Tartiflette',
+    'Tempura',
+    'tiramisu',
+    'Tteokbokki_food',
+    'tuna_tartare',
+    'Udon',
+    'unni_appam',
+    'waffles',
+    'Yakitori',
+    'Yangnyeom_Tongdak_food'
+)
+
+
+
+model_path = Path('food-recognizer-r50-v0.pkl')
+model = load_learner(model_path)
+
+def recognize_image(image):
+  pred, idx, probs = model.predict(image)
+  print(pred, probs)
+  return dict(zip(food_labels, map(float, probs)))
+
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label(num_top_classes=5)
+examples = [
+    'test1.jpg',
+    'test2.jpg',
+    'test3.jpg',
+    'test4.jpg',
+    'test5.jpg',
+    'test6.jpg'
+    ]
+
+iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
+iface.launch(inline=False)