Spaces:
Runtime error
Runtime error
File size: 4,985 Bytes
ba67ea5 d03bcd6 ba67ea5 8204bc2 7558946 8204bc2 ba67ea5 8204bc2 7558946 ba67ea5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
from fastai.vision.all import *
import gradio as gr
def get_label(filename):
folder_name = parent_label(filename)
return folder_name.split('.')[1]
learn = load_learner('model.pkl')
categories = (
'Acadian_Flycatcher', 'American_Crow', 'American_Goldfinch', 'American_Pipit',
'American_Redstart', 'American_Three_toed_Woodpecker', 'Anna_Hummingbird',
'Artic_Tern', 'Baird_Sparrow', 'Baltimore_Oriole', 'Bank_Swallow', 'Barn_Swallow',
'Bay_breasted_Warbler', 'Belted_Kingfisher', 'Bewick_Wren', 'Black_Tern',
'Black_and_white_Warbler', 'Black_billed_Cuckoo', 'Black_capped_Vireo',
'Black_footed_Albatross', 'Black_throated_Blue_Warbler', 'Black_throated_Sparrow',
'Blue_Grosbeak', 'Blue_Jay', 'Blue_headed_Vireo', 'Blue_winged_Warbler',
'Boat_tailed_Grackle', 'Bobolink', 'Bohemian_Waxwing', 'Brandt_Cormorant',
'Brewer_Blackbird', 'Brewer_Sparrow', 'Bronzed_Cowbird', 'Brown_Creeper',
'Brown_Pelican', 'Brown_Thrasher', 'Cactus_Wren', 'California_Gull',
'Canada_Warbler', 'Cape_Glossy_Starling', 'Cape_May_Warbler', 'Cardinal',
'Carolina_Wren', 'Caspian_Tern', 'Cedar_Waxwing', 'Cerulean_Warbler',
'Chestnut_sided_Warbler', 'Chipping_Sparrow', 'Chuck_will_Widow', 'Clark_Nutcracker',
'Clay_colored_Sparrow', 'Cliff_Swallow', 'Common_Raven', 'Common_Tern',
'Common_Yellowthroat', 'Crested_Auklet', 'Dark_eyed_Junco', 'Downy_Woodpecker',
'Eared_Grebe', 'Eastern_Towhee', 'Elegant_Tern', 'European_Goldfinch',
'Evening_Grosbeak', 'Field_Sparrow', 'Fish_Crow', 'Florida_Jay', 'Forsters_Tern',
'Fox_Sparrow', 'Frigatebird', 'Gadwall', 'Geococcyx', 'Glaucous_winged_Gull',
'Golden_winged_Warbler', 'Grasshopper_Sparrow', 'Gray_Catbird', 'Gray_Kingbird',
'Gray_crowned_Rosy_Finch', 'Great_Crested_Flycatcher', 'Great_Grey_Shrike',
'Green_Jay', 'Green_Kingfisher', 'Green_Violetear', 'Green_tailed_Towhee',
'Groove_billed_Ani', 'Harris_Sparrow', 'Heermann_Gull', 'Henslow_Sparrow',
'Herring_Gull', 'Hooded_Merganser', 'Hooded_Oriole', 'Hooded_Warbler',
'Horned_Grebe', 'Horned_Lark', 'Horned_Puffin', 'House_Sparrow', 'House_Wren',
'Indigo_Bunting', 'Ivory_Gull', 'Kentucky_Warbler', 'Laysan_Albatross',
'Lazuli_Bunting', 'Le_Conte_Sparrow', 'Least_Auklet', 'Least_Flycatcher',
'Least_Tern', 'Lincoln_Sparrow', 'Loggerhead_Shrike', 'Long_tailed_Jaeger',
'Louisiana_Waterthrush', 'Magnolia_Warbler', 'Mallard', 'Mangrove_Cuckoo',
'Marsh_Wren', 'Mockingbird', 'Mourning_Warbler', 'Myrtle_Warbler', 'Nashville_Warbler',
'Nelson_Sharp_tailed_Sparrow', 'Nighthawk', 'Northern_Flicker', 'Northern_Fulmar',
'Northern_Waterthrush', 'Olive_sided_Flycatcher', 'Orange_crowned_Warbler',
'Orchard_Oriole', 'Ovenbird', 'Pacific_Loon', 'Painted_Bunting', 'Palm_Warbler',
'Parakeet_Auklet', 'Pelagic_Cormorant', 'Philadelphia_Vireo', 'Pied_Kingfisher',
'Pied_billed_Grebe', 'Pigeon_Guillemot', 'Pileated_Woodpecker', 'Pine_Grosbeak',
'Pine_Warbler', 'Pomarine_Jaeger', 'Prairie_Warbler', 'Prothonotary_Warbler',
'Purple_Finch', 'Red_bellied_Woodpecker', 'Red_breasted_Merganser',
'Red_cockaded_Woodpecker', 'Red_eyed_Vireo', 'Red_faced_Cormorant',
'Red_headed_Woodpecker', 'Red_legged_Kittiwake', 'Red_winged_Blackbird',
'Rhinoceros_Auklet', 'Ring_billed_Gull', 'Ringed_Kingfisher', 'Rock_Wren',
'Rose_breasted_Grosbeak', 'Ruby_throated_Hummingbird', 'Rufous_Hummingbird',
'Rusty_Blackbird', 'Sage_Thrasher', 'Savannah_Sparrow', 'Sayornis',
'Scarlet_Tanager', 'Scissor_tailed_Flycatcher', 'Scott_Oriole',
'Seaside_Sparrow', 'Shiny_Cowbird', 'Slaty_backed_Gull', 'Song_Sparrow',
'Sooty_Albatross', 'Spotted_Catbird', 'Summer_Tanager', 'Swainson_Warbler',
'Tennessee_Warbler', 'Tree_Sparrow', 'Tree_Swallow', 'Tropical_Kingbird',
'Vermilion_Flycatcher', 'Vesper_Sparrow', 'Warbling_Vireo', 'Western_Grebe',
'Western_Gull', 'Western_Meadowlark', 'Western_Wood_Pewee', 'Whip_poor_Will',
'White_Pelican', 'White_breasted_Kingfisher', 'White_breasted_Nuthatch',
'White_crowned_Sparrow', 'White_eyed_Vireo', 'White_necked_Raven',
'White_throated_Sparrow', 'Wilson_Warbler', 'Winter_Wren', 'Worm_eating_Warbler',
'Yellow_Warbler', 'Yellow_bellied_Flycatcher', 'Yellow_billed_Cuckoo',
'Yellow_breasted_Chat', 'Yellow_headed_Blackbird', 'Yellow_throated_Vireo'
)
def classify_image(img):
pred, idx, probs = learn.predict(img)
return dict(zip(categories, map(float, probs)))
image = gr.Image(shape=(192,192))
label = gr.Label()
examples = ['examples/acadian_flycatcher.jpeg', 'examples/pacific_loon.jpg', 'examples/yellow_throated_vireo.jpeg']
title='North America Bird Classifier'
description='A north america bird classifier trained on the Caltech-UCSD Birds-200-2011 with fastai.'
iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, title=title,description=description, examples=examples)
iface.launch(inline=False) |