Spaces:
Sleeping
Sleeping
all
Browse files
app.py
ADDED
|
@@ -0,0 +1,269 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import gradio as gr
|
| 2 |
+
from fastai.vision.all import load_learner
|
| 3 |
+
|
| 4 |
+
|
| 5 |
+
food_labels = (
|
| 6 |
+
'adhirasam',
|
| 7 |
+
'aloo_gobi',
|
| 8 |
+
'aloo_matar',
|
| 9 |
+
'aloo_methi',
|
| 10 |
+
'aloo_shimla_mirch',
|
| 11 |
+
'aloo_tikki',
|
| 12 |
+
'anarsa',
|
| 13 |
+
'apple_pie',
|
| 14 |
+
'ariselu',
|
| 15 |
+
'baby_back_ribs',
|
| 16 |
+
'Baden_Brägele_with_Bibliskäs',
|
| 17 |
+
'baklava',
|
| 18 |
+
'bandar_laddu',
|
| 19 |
+
'basundi',
|
| 20 |
+
'beef_carpaccio',
|
| 21 |
+
'beef_tartare',
|
| 22 |
+
'beet_salad',
|
| 23 |
+
'beignets',
|
| 24 |
+
'bhatura',
|
| 25 |
+
'bhindi_masala',
|
| 26 |
+
'Bibimbap_food',
|
| 27 |
+
'biryani',
|
| 28 |
+
'Blanquette_de_Veau',
|
| 29 |
+
'Boeuf_Bourguignon',
|
| 30 |
+
'boondi',
|
| 31 |
+
'Bossam_food',
|
| 32 |
+
'Bouillabaisse',
|
| 33 |
+
'Braised_Beef_Rheinland_style',
|
| 34 |
+
'Braunkohl_and_Bregenwurst',
|
| 35 |
+
'bread_pudding',
|
| 36 |
+
'breakfast_burrito',
|
| 37 |
+
'bruschetta',
|
| 38 |
+
'Budae_Jjigae_food',
|
| 39 |
+
'Bulgogi_food',
|
| 40 |
+
'butter_chicken',
|
| 41 |
+
'Butter_Noodles',
|
| 42 |
+
'caesar_salad',
|
| 43 |
+
'cannoli',
|
| 44 |
+
'caprese_salad',
|
| 45 |
+
'carrot_cake',
|
| 46 |
+
'Cassoulet',
|
| 47 |
+
'ceviche',
|
| 48 |
+
'chak_hao_kheer',
|
| 49 |
+
'cham_cham',
|
| 50 |
+
'chana_masala',
|
| 51 |
+
'chapati',
|
| 52 |
+
'cheesecake',
|
| 53 |
+
'cheese_plate',
|
| 54 |
+
'chhena_kheeri',
|
| 55 |
+
'chicken_quesadilla',
|
| 56 |
+
'chicken_razala',
|
| 57 |
+
'chicken_tikka',
|
| 58 |
+
'chicken_tikka_masala',
|
| 59 |
+
'chicken_wings',
|
| 60 |
+
'chikki',
|
| 61 |
+
'chocolate_cake',
|
| 62 |
+
'chocolate_mousse',
|
| 63 |
+
'Chuncheon_Dakgalbi_food',
|
| 64 |
+
'churros',
|
| 65 |
+
'clam_chowder',
|
| 66 |
+
'club_sandwich',
|
| 67 |
+
'Coquilles_Saint_Jacques',
|
| 68 |
+
'Coq_au_Vin',
|
| 69 |
+
'crab_cakes',
|
| 70 |
+
'creme_brulee',
|
| 71 |
+
'croque_madame',
|
| 72 |
+
'Crème_brûlée',
|
| 73 |
+
'Crêpes',
|
| 74 |
+
'cup_cakes',
|
| 75 |
+
'daal_baati_churma',
|
| 76 |
+
'daal_puri',
|
| 77 |
+
'dal_makhani',
|
| 78 |
+
'dal_tadka',
|
| 79 |
+
'deviled_eggs',
|
| 80 |
+
'dharwad_pedha',
|
| 81 |
+
'donuts',
|
| 82 |
+
'doodhpak',
|
| 83 |
+
'double_ka_meetha',
|
| 84 |
+
'Dumplings',
|
| 85 |
+
'dum_aloo',
|
| 86 |
+
'Döner_with_Curried_Sausage',
|
| 87 |
+
'edamame',
|
| 88 |
+
'eggs_benedict',
|
| 89 |
+
'Escargots',
|
| 90 |
+
'falafel',
|
| 91 |
+
'filet_mignon',
|
| 92 |
+
'fish_and_chips',
|
| 93 |
+
'foie_gras',
|
| 94 |
+
'Fondue_Savoyarde',
|
| 95 |
+
'Frankfurter_Green_Sauce',
|
| 96 |
+
'french_fries',
|
| 97 |
+
'french_onion_soup',
|
| 98 |
+
'french_toast',
|
| 99 |
+
'fried_calamari',
|
| 100 |
+
'fried_rice',
|
| 101 |
+
'frozen_yogurt',
|
| 102 |
+
'gajar_ka_halwa',
|
| 103 |
+
'garlic_bread',
|
| 104 |
+
'gavvalu',
|
| 105 |
+
'German_Bread_Rolls',
|
| 106 |
+
'ghevar',
|
| 107 |
+
'Gimbap_food',
|
| 108 |
+
'gnocchi',
|
| 109 |
+
'greek_salad',
|
| 110 |
+
'grilled_cheese_sandwich',
|
| 111 |
+
'grilled_salmon',
|
| 112 |
+
'guacamole',
|
| 113 |
+
'gulab_jamun',
|
| 114 |
+
'gyoza',
|
| 115 |
+
'Gyudon',
|
| 116 |
+
'hamburger',
|
| 117 |
+
'Hamburger_Labskaus',
|
| 118 |
+
'Hamburger_Pannfisch',
|
| 119 |
+
'hot_and_sour_soup',
|
| 120 |
+
'hot_dog',
|
| 121 |
+
'huevos_rancheros',
|
| 122 |
+
'hummus',
|
| 123 |
+
'ice_cream',
|
| 124 |
+
'imarti',
|
| 125 |
+
'jalebi',
|
| 126 |
+
'kachori',
|
| 127 |
+
'kadai_paneer',
|
| 128 |
+
'kadhi_pakoda',
|
| 129 |
+
'kajjikaya',
|
| 130 |
+
'kakinada_khaja',
|
| 131 |
+
'kalakand',
|
| 132 |
+
'Kalguksu_food',
|
| 133 |
+
'karela_bharta',
|
| 134 |
+
'Kare_raisu',
|
| 135 |
+
'Katsudon',
|
| 136 |
+
'Kimchi_Jjigae_food',
|
| 137 |
+
'kofta',
|
| 138 |
+
'kuzhi_paniyaram',
|
| 139 |
+
'Königsberger_Meatballs',
|
| 140 |
+
'lasagna',
|
| 141 |
+
'lassi',
|
| 142 |
+
'ledikeni',
|
| 143 |
+
'litti_chokha',
|
| 144 |
+
'lobster_bisque',
|
| 145 |
+
'lobster_roll_sandwich',
|
| 146 |
+
'lyangcha',
|
| 147 |
+
'maach_jhol',
|
| 148 |
+
'macaroni_and_cheese',
|
| 149 |
+
'macarons',
|
| 150 |
+
'makki_di_roti_sarson_da_saag',
|
| 151 |
+
'malapua',
|
| 152 |
+
'Manduguk_food',
|
| 153 |
+
'misi_roti',
|
| 154 |
+
'Miso_soup',
|
| 155 |
+
'misti_doi',
|
| 156 |
+
'modak',
|
| 157 |
+
'Moules_Marinières',
|
| 158 |
+
'mussels',
|
| 159 |
+
'mysore_pak',
|
| 160 |
+
'naan',
|
| 161 |
+
'nachos',
|
| 162 |
+
'navrattan_korma',
|
| 163 |
+
'Ojingeo_Bokkeum_food',
|
| 164 |
+
'Okonomiyaki',
|
| 165 |
+
'omelette',
|
| 166 |
+
'Omurice',
|
| 167 |
+
'Onigiri',
|
| 168 |
+
'onion_rings',
|
| 169 |
+
'oysters',
|
| 170 |
+
'pad_thai',
|
| 171 |
+
'paella',
|
| 172 |
+
'Pajeon_food',
|
| 173 |
+
'palak_paneer',
|
| 174 |
+
'pancakes',
|
| 175 |
+
'paneer_butter_masala',
|
| 176 |
+
'panna_cotta',
|
| 177 |
+
'peking_duck',
|
| 178 |
+
'phirni',
|
| 179 |
+
'pho',
|
| 180 |
+
'Pissaladière',
|
| 181 |
+
'pithe',
|
| 182 |
+
'pizza',
|
| 183 |
+
'poha',
|
| 184 |
+
'poornalu',
|
| 185 |
+
'pootharekulu',
|
| 186 |
+
'pork_chop',
|
| 187 |
+
'Pork_Knuckle_with_Sauerkraut',
|
| 188 |
+
'Potato_Fritters_or_Potato_Pancakes_Kartoffelpuffer_',
|
| 189 |
+
'Potato_Salad',
|
| 190 |
+
'poutine',
|
| 191 |
+
'Pretzen_from_Bavaria',
|
| 192 |
+
'prime_rib',
|
| 193 |
+
'pulled_pork_sandwich',
|
| 194 |
+
'Pumpernickel_from_Lower_Saxony',
|
| 195 |
+
'qubani_ka_meetha',
|
| 196 |
+
'rabri',
|
| 197 |
+
'Ramen',
|
| 198 |
+
'rasgulla',
|
| 199 |
+
'ras_malai',
|
| 200 |
+
'ravioli',
|
| 201 |
+
'red_velvet_cake',
|
| 202 |
+
'risotto',
|
| 203 |
+
'Salade_niçoise',
|
| 204 |
+
'Samgyetang_food',
|
| 205 |
+
'samosa',
|
| 206 |
+
'sandesh',
|
| 207 |
+
'sashimi',
|
| 208 |
+
'scallops',
|
| 209 |
+
'seaweed_salad',
|
| 210 |
+
'Shabu_Shabu',
|
| 211 |
+
'shankarpali',
|
| 212 |
+
'sheera',
|
| 213 |
+
'sheer_korma',
|
| 214 |
+
'shrikhand',
|
| 215 |
+
'shrimp_and_grits',
|
| 216 |
+
'Soba',
|
| 217 |
+
'sohan_halwa',
|
| 218 |
+
'sohan_papdi',
|
| 219 |
+
'Soufflé',
|
| 220 |
+
'Soupe_à_L_oignon',
|
| 221 |
+
'spaghetti_bolognese',
|
| 222 |
+
'spaghetti_carbonara',
|
| 223 |
+
'spring_rolls',
|
| 224 |
+
'steak',
|
| 225 |
+
'Steak_Tartare',
|
| 226 |
+
'strawberry_shortcake',
|
| 227 |
+
'Sundae_food',
|
| 228 |
+
'sushi',
|
| 229 |
+
'Sushi_Sashimi',
|
| 230 |
+
'sutar_feni',
|
| 231 |
+
'tacos',
|
| 232 |
+
'takoyaki',
|
| 233 |
+
'Tarte_Flambée_with_a_French_Influence',
|
| 234 |
+
'Tarte_Tatin',
|
| 235 |
+
'Tartiflette',
|
| 236 |
+
'Tempura',
|
| 237 |
+
'tiramisu',
|
| 238 |
+
'Tteokbokki_food',
|
| 239 |
+
'tuna_tartare',
|
| 240 |
+
'Udon',
|
| 241 |
+
'unni_appam',
|
| 242 |
+
'waffles',
|
| 243 |
+
'Yakitori',
|
| 244 |
+
'Yangnyeom_Tongdak_food'
|
| 245 |
+
)
|
| 246 |
+
|
| 247 |
+
|
| 248 |
+
|
| 249 |
+
|
| 250 |
+
model = load_learner('food-recognizer-r50-v0.pkl')
|
| 251 |
+
|
| 252 |
+
def recognize_image(image):
|
| 253 |
+
pred, idx, probs = model.predict(image)
|
| 254 |
+
print(pred, probs)
|
| 255 |
+
return dict(zip(food_labels, map(float, probs)))
|
| 256 |
+
|
| 257 |
+
image = gr.inputs.Image(shape=(192,192))
|
| 258 |
+
label = gr.outputs.Label(num_top_classes=5)
|
| 259 |
+
examples = [
|
| 260 |
+
'test1.jpg',
|
| 261 |
+
'test2.jpg',
|
| 262 |
+
'test3.jpg',
|
| 263 |
+
'test4.jpg',
|
| 264 |
+
'test5.jpg',
|
| 265 |
+
'test6.jpg'
|
| 266 |
+
]
|
| 267 |
+
|
| 268 |
+
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
|
| 269 |
+
iface.launch(inline=False)
|