git push

app.py

@@ -1,9 +1,246 @@
+from fastai.vision.widgets import *
 import gradio as gr
+import numpy as np
+from operator import itemgetter
+from fastai.vision.all import *
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+learn = load_learner("./plants_10_epochs.pkl")
 
+categories = ('Salvia officinalis',
+ 'Allium sativum',
+ 'Coriandrum sativum',
+ 'Myrica gale',
+ 'Curcuma longa',
+ 'Xylopia aethiopica',
+ 'Thymus vulgaris',
+ 'Tamarindus indica',
+ 'Artemisia pallens',
+ 'Lavandula latifolia',
+ 'Porophyllum ruderale',
+ 'Achillea millefolium',
+ 'Tagetes minuta',
+ 'Citrus aurantifolia',
+ 'Mentha longifolia',
+ 'Dictamnus albus',
+ 'Cuminum cyminum',
+ 'Rosa centifolia',
+ 'Pelargonium graveolens',
+ 'Rumex acetosa',
+ 'Rosa spinossima',
+ 'Lavandula stoechas',
+ 'Crocus sativus',
+ 'Rosmarinus officinalis',
+ 'Agastache foeniculum',
+ 'Nigella sativa',
+ 'Rosa rugosa',
+ 'Trigonella caerulea',
+ 'Herbs and Spices.',
+ 'Capparis spinosa',
+ 'Piper longum',
+ 'Lippia graveolens',
+ 'Helichrysum italicum',
+ 'Illicium verum',
+ 'Mentha aquatica',
+ 'Carum carvi',
+ 'Angelica atropurpurea',
+ 'Schmeil-Fitschen, Flora von Deutschland',
+ 'Agastache mexicana',
+ 'Kräuter und Gewürze.',
+ 'Aloysia citrodora',
+ 'Amomum subulatum',
+ 'Rungia klossii',
+ 'Dysphania ambrosioides',
+ 'Rosa damascena',
+ 'Filipendula ulmaria',
+ 'Teucrium scorodonia',
+ 'Rosa multiflora',
+ 'Pimpinella anisum',
+ 'Piper cubeba',
+ 'Myristica fragrans',
+ 'Origanum majorana',
+ 'Rosa eglanteria',
+ 'Acorus gramineus',
+ 'Thymus serpyllum',
+ 'Solanum lycopersicum',
+ 'Kaempferia galanga',
+ 'Monarda citriodora',
+ 'Sesamum indicum',
+ 'Betonica officinalis',
+ 'Mentha sativa',
+ 'Artemisia vulgaris',
+ 'Melissa officinalis',
+ 'Cinnamomum cassia',
+ 'Origanum vulgare',
+ 'Cryptotaenia japonica',
+ 'Glechoma hederacea',
+ 'Stachys affinis',
+ 'Tasmannia lanceolata',
+ 'Hyptis suaveolens',
+ 'Nigella damascena',
+ 'Tsuga canadensis',
+ 'Teucrium chamaedrys',
+ 'Solanum melongena',
+ 'Cymbopogon citratus',
+ 'Vanilla tahitensis',
+ 'Satureja montana',
+ 'Cananga odorata',
+ 'Allium cepa',
+ 'Eruca sativa',
+ 'Solidago odora',
+ 'Nasturtium officinale',
+ 'Anthriscus cerefolium',
+ 'Cymbopogon flexuosus',
+ 'Vanilla planifolia',
+ 'Matricaria chamomilla',
+ 'Apium graveolens',
+ 'Taraxacum officinale',
+ 'Chrysanthemum vulgare',
+ 'Salvia sclarea',
+ 'Oxalis acetosella',
+ 'Allium scorodoprasum',
+ 'Elettaria cardamomum',
+ 'Dianthus caryophyllus',
+ 'Allium porrum',
+ 'Allium schoenoprasum',
+ 'Syzygium aromaticum',
+ 'Lepidium latifolium',
+ 'Curcuma amada',
+ 'Plectranthus amboinicus',
+ 'Monarda fistulosa',
+ 'Salvia elegans',
+ 'Hedeoma pulegioides',
+ 'Levisticum officinale',
+ 'Monodora myristica',
+ 'Thymbra capitata',
+ 'Smilax regelii',
+ 'Rhus coriaria',
+ 'Laserpitium siler',
+ 'Trigonella foenum-graecum',
+ 'Calendula officinalis',
+ 'Zanthoxylum acanthopodium',
+ 'Syzygium polyanthum',
+ 'Filipendula vulgaris',
+ 'Capsicum baccatum',
+ 'Picea glauca',
+ 'Eryngium foetidum',
+ 'Piper nigrum',
+ 'Bergera koenigii',
+ 'Gaultheria procumbens',
+ 'Myrrhis odorata',
+ 'Foeniculum vulgare',
+ 'Citrus hystrix',
+ 'Mentha rotundifolia',
+ 'Pastinaca sativa',
+ 'Capsicum frutescens',
+ 'Vetiveria zizanioides',
+ 'Mentha pulegium',
+ 'Alpinia galanga',
+ 'Allium fistulosum',
+ 'Sinapis alba',
+ 'Salvia pratensis',
+ 'Limnophila aromatica',
+ 'Myrtus communis',
+ 'Rosa canina',
+ 'Cunila origanoides',
+ 'Zingiber officinale',
+ 'Mentha suaveolens',
+ 'Boronia megastigma',
+ 'Portulaca sativa',
+ 'Perilla frutescens',
+ 'Nepeta cataria',
+ 'Smallanthus sonchifolia',
+ 'Elsholtzia ciliata',
+ 'Pimpinella saxifraga',
+ 'Cochlearia officinalis',
+ 'Allium ursinum',
+ 'Symphytum officinale',
+ 'Capsicum annuum var. acuminatum',
+ 'Mentha arvensis',
+ 'Borago officinalis',
+ 'Sanguisorba officinalis',
+ 'Origanum dictamnus',
+ 'Agastache rugosa',
+ 'Piper auritum',
+ 'Armoracia lapathifolia',
+ 'Aframomum melegueta',
+ 'Meum athamanticum',
+ 'Humulus lupulus',
+ 'Thymus mastichina',
+ 'Monarda didyma',
+ 'Picea mariana',
+ 'Satureja hortensis',
+ 'Alliaria petiolata',
+ 'Chrysanthemum balsamita',
+ 'Hyssopus officinalis',
+ 'Ocimum viride',
+ 'Anethum graveolens',
+ 'Valeriana officinalis',
+ 'Anthemis nobilis',
+ 'Artemisia dracunculus',
+ 'Origanum syriacum',
+ 'Smilax aristolochiifolia',
+ 'Ferula asafoetida',
+ 'Marrubium vulgare',
+ 'Bixa orellana',
+ 'Eutrema japonica',
+ 'Cinnamomum verum',
+ 'Pimpinella major',
+ 'Achillea decolorans',
+ 'Schinus molle',
+ 'Hierochloe odorata',
+ 'Viola odorata',
+ 'Aniba rosaeodora',
+ 'Zanthoxylum piperitum',
+ 'Capsicum chinense',
+ 'Galium verum',
+ 'Allium tuberosum',
+ 'Schinus terebinthifolius',
+ 'Rosa gallica',
+ 'Laurus nobilis',
+ 'Acorus calamus',
+ 'Rumex scutatus',
+ 'Capsicum annuum',
+ 'Capsicum pubescens',
+ 'Mentha spicata',
+ 'Mentha nemorosa',
+ 'Lepidium sativum',
+ 'Oxalis violacea',
+ 'Brassica nigra',
+ 'Juniperus communis',
+ 'Parkia biglobosa',
+ 'Artemisia abrotanum',
+ 'Dipteryx odorata',
+ 'Sassafras albidum',
+ 'Tropaeolum majus',
+ 'Lavandula angustifolia',
+ 'Acinos alpinus',
+ 'Artemisia absinthium',
+ 'Angelica archangelica',
+ 'Ruta graveolens',
+ 'Bunium persicum',
+ 'Vanilla pompona',
+ 'Erigeron canadensis',
+ 'Curcuma zedoaria',
+ 'Pimenta dioica',
+ 'Mentha citrata',
+ 'Sanguisorba minor',
+ 'Rosa moschata',
+ 'Galium odoratum',
+ 'Ocimum basilicum',
+ 'Cannabis sativa',
+ 'Trachyspermum ammi',
+ 'Persicaria hydropiper')
 
+def classify_image(img):
+    pred, idx, probs = learn.predict(img)
+    dict_all = dict(zip(categories, map(float,probs)))
+    n = 5
+    dict_n = dict(sorted(dict_all.items(), key=itemgetter(1), reverse=True)[:n])
+    return dict_n
+
+image = gr.inputs.Image(shape=(192,192))
+label=gr.outputs.Label()
+
+intf=gr.Interface(fn=classify_image, inputs=image, outputs=label)
+intf.launch(inline=False)