Update app.py

app.py

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