sanjid commited on
Commit
55cf03f
·
1 Parent(s): c21e91e
Files changed (1) hide show
  1. app.py +0 -269
app.py DELETED
@@ -1,269 +0,0 @@
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)