Spaces:
Build error
Build error
| import gradio as gr | |
| from fastai.vision.all import * | |
| import skimage | |
| learn = load_learner('export_ninjago_classifier.pkl') | |
| labels = ['Acidicus (Serpentine)', | |
| 'Acronix (Vermillion)', | |
| 'Akita (Never-Realm inhabitants)', | |
| 'Antonia (Recurring allies)', | |
| 'Ash (Elemental Masters)', | |
| 'Aspheera (Pyro Vipers)', | |
| 'Bansha (Ghosts)', | |
| 'Benthomaar (Merlopians)', | |
| 'Blunck (Vermillion)', | |
| 'Bolobo (Elemental Masters)', | |
| 'Bucko (Sky Pirates)', | |
| 'Captain Soto (Citizens of Ninjago City)', | |
| 'Chamille (Elemental Masters)', | |
| 'Char (Pyro Vipers)', | |
| 'Chew Toy (Dragon Hunters)', | |
| 'Chief Mammatus (The Island inhabitants)', | |
| 'Chope (Anacondrai Cultists)', | |
| 'Clancee (Sky Pirates)', | |
| 'Clouse (Anacondrai Cultists)', | |
| 'Clutch Powers (Recurring allies)', | |
| 'Cole (Main Characters)', | |
| 'Cyren (Sky Pirates)', | |
| 'Cyrus Borg (Recurring allies)', | |
| 'Daddy No Legs (Dragon Hunters)', | |
| 'Dareth (Recurring allies)', | |
| 'Dogshank (Sky Pirates)', | |
| 'Doubloon (Sky Pirates)', | |
| 'Dr. Julien (Family members)', | |
| 'Edna Walker (Family members)', | |
| 'Ed Walker (Family members)', | |
| 'Eyezor (Anacondrai Cultists)', | |
| 'Faith (Dragon Hunters)', | |
| 'Fangtom (Serpentine)', | |
| 'Flintlocke (Sky Pirates)', | |
| 'Fred Finely (Citizens of Ninjago City)', | |
| 'Fuchsia Ninja (New Ninja)', | |
| 'Fugi-Dove (Citizens of Ninjago City)', | |
| 'General Arcturus (Serpentine)', | |
| 'General Cryptor (Nindroids)', | |
| 'General Kozu (Stone Army)', | |
| 'General Vex (Never-Realm inhabitants)', | |
| 'Ghoultar (Ghosts)', | |
| 'Gleck (Shintaro Kingdom inhabitants)', | |
| 'Glutinous (Merlopians)', | |
| 'Gravis (Elemental Masters)', | |
| 'Griffin Turner (Elemental Masters)', | |
| 'Hounddog McBrag (Citizens of Ninjago City)', | |
| 'Jacob Pevsner (Elemental Masters)', | |
| 'Jay (Main Characters)', | |
| 'Jet Jack (Dragon Hunters)', | |
| 'Kai (Main Characters)', | |
| 'Kapau (Anacondrai Cultists)', | |
| 'Karlof (Elemental Masters)', | |
| 'Khanjikhan (Sky Pirates)', | |
| 'Killow (Sons of Garmadon)', | |
| 'King Trimaar (Merlopians)', | |
| 'King Vangelis_The Skull Sorcerer (Shintaro Kingdom inhabitants)', | |
| 'Kruncha (Skulkin)', | |
| 'Krux (Vermillion)', | |
| 'Lloyd Garmadon (Main Characters)', | |
| 'Lord Garmadon (Main Characters)', | |
| 'Luke Cunningham (Sons of Garmadon)', | |
| 'Machia (Vermillion)', | |
| 'Master Chen (Anacondrai Cultists)', | |
| 'Master Wu (Main Characters)', | |
| 'Maya (Family members)', | |
| 'Milton Dyer (Prime Empire inhabitants)', | |
| 'Min-Droid (Nindroids)', | |
| 'Misako (Family members)', | |
| 'Mistaké (Recurring allies)', | |
| 'Mister F (Council of the Crystal King)', | |
| 'Monkey Wretch (Sky Pirates)', | |
| 'Morro (Ghosts)', | |
| 'Mr. E (Sons of Garmadon)', | |
| 'Murt (Shintaro Kingdom inhabitants)', | |
| 'Muzzle (Dragon Hunters)', | |
| 'Nadakhan (Sky Pirates)', | |
| 'Nelson (Recurring allies)', | |
| 'Neuro (Elemental Masters)', | |
| 'Nuckal (Skulkin)', | |
| 'Nya (Main Characters)', | |
| 'Officer Toque (Recurring allies)', | |
| 'Okino (Prime Empire inhabitants)', | |
| 'Orange Ninja (New Ninja)', | |
| 'Paleman (Elemental Masters)', | |
| 'Patty Keys (Citizens of Ninjago City)', | |
| 'Pink Ninja (New Ninja)', | |
| 'P.I.X.A.L. (Main Characters)', | |
| 'Police Commissioner (Recurring allies)', | |
| 'Postman (Citizens of Ninjago City)', | |
| 'Prince Kalmaar (Merlopians)', | |
| 'Princess Harumi (Sons of Garmadon)', | |
| 'Princess Vania (Shintaro Kingdom inhabitants)', | |
| 'Pythor P. Chumsworth (Serpentine)', | |
| 'Raggmunk (Vermillion)', | |
| 'Ray (Family members)', | |
| 'Ronin (Recurring allies)', | |
| 'Rufus MacAllister (Citizens of Ninjago City)', | |
| 'Samukai (Skulkin)', | |
| 'Scott (Prime Empire inhabitants)', | |
| 'Sensei_Master Kodokuna Yang (Ghosts)', | |
| 'Shade (Elemental Masters)', | |
| 'Skales (Serpentine)', | |
| 'Skalidor (Serpentine)', | |
| 'Skylor Chen (Elemental Masters)', | |
| 'Slithraa (Serpentine)', | |
| 'Soul Archer (Ghosts)', | |
| 'Squiffy (Sky Pirates)', | |
| 'Sushimi (Prime Empire inhabitants)', | |
| 'The First Spinjitzu Master (Family members)', | |
| 'The Great Devourer (Serpentine)', | |
| 'The Ice Emperor (Never-Realm inhabitants)', | |
| 'The Iron Baron (Dragon Hunters)', | |
| 'The Mechanic (Prime Empire inhabitants)', | |
| 'The Omega (Oni)', | |
| 'The Overlord (Stone Army)', | |
| 'The Preeminent (Ghosts)', | |
| 'Timothy "Twitchy Tim" Batterson (The Island inhabitants)', | |
| 'Tox (Elemental Masters)', | |
| 'Ultra Violet (Sons of Garmadon)', | |
| 'Unagami (Prime Empire inhabitants)', | |
| 'Wrayth (Ghosts)', | |
| 'Wyplash (Skulkin)', | |
| 'Zane (Main Characters)', | |
| 'Zugu (Anacondrai Cultists)'] | |
| def predict(img): | |
| img = PILImage.create(img) | |
| pred, pred_idx, probs = learn.predict(img) | |
| return {labels[i]: float(probs[i]) for i in range(len(labels))} | |
| title = "Ninjago Character Classifier" | |
| description = "Guesses the name of the Ninjago characters. Created from the fastai demo for Gradio and HuggingFace Spaces." | |
| #article = "<p style='text-align: center'><a href='https://tmabraham.github.io/blog/gradio_hf_spaces_tutorial' target='_blank'>Blog post</a></p>" | |
| examples = ['Lloyd.jpg', 'Cole.png'] | |
| interpretation = 'default' | |
| enable_queue = True | |
| gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(192, 192)), outputs=gr.outputs.Label(num_top_classes=3), title=title, | |
| description=description, examples=examples, interpretation=interpretation, enable_queue=enable_queue).launch() | |